Hi, isn't this issue relevant to your code? Also this is for the same thing.
I would make a PR but unfortunately I don't know enough about Emacs theming.