programs.zsh.syntaxHighlighting incorrect behaviour with ´set -o vi´ or vi plugin installed

[Kazimazi]

Describe the bug
While I type in insert mode syntax highlighting won't appear. If I press esc to enter normal mode or press backspace to delete a char, keywords get highlighted as expected. vi-mode is not mandatory we could achieve the bug with $ set -o vi

To Reproduce
Steps to reproduce the behavior:
Have this config:

programs.zsh = {
  enable = true;
  syntaxHighlighting.enable = true;
  ohMyZsh = {
    enable = true;
    plugins = [ "vi-mode" ];
  };
};

1. start typing for example: $ sudo nixos-reb. The line won't have any colors to it till...
2. press backspace, or enter normal mode

Expected behavior
Syntax gets refreshed.

Metadata
Please run nix run nixpkgs.nix-info -c nix-info -m and paste the result.

 - system: `"x86_64-linux"`
 - host os: `Linux 5.3.10, NixOS, 19.09.1223.cb2cdab7136 (Loris)`
 - multi-user?: `yes`
 - sandbox: `yes`
 - version: `nix-env (Nix) 2.3`
 - channels(root): `"nixos-19.09.1223.cb2cdab7136, nixos-unstable-20.03pre200231.7827d3f4497"`
 - channels(kazimazi): `""`
 - nixpkgs: `/nix/var/nix/profiles/per-user/root/channels/nixos`