Found out you need to switch UI\Units too (for the small icon).

Here is the fix (.rar - readme inside):
Komatai Fix