Temp fix for no highlight for code selection