Fixy Frosted Flakes

Fixes a bug with the previous bug fix.

This bugfix I removed was actually a bug fix to fix a bug in VSCode. That bug had been removed in VSCode Insiders, hence why I didn't notice it while testing! I've now reverted to using VSCode all the time. :)