Overleaf in VIM mode not allowing caps lock to return to normal mode

So im trying to get started using overleaf as its recommended on one of my courses, but for some reason Overleafs VIM mode is not detecting the escape key properly. Im on Ubuntu 20.04 and ive swapped escape and caps lock using setxkbmap -option caps:swapescape. But if im in Overleaf insert mode, pressing caps lock does not return me to Normal mode. Only way to go back to normal mode is by pressing the esc key, which also toggles caps lock on. Its very frustrating and not usable.

Any ideas what I could do to fix it?