VSCode as EDITOR doesnt seem to work

[Tautvydas]

~ echo $EDITOR
code --wait

Pressing enter on a file does not seem to open VSCode