Fix shell highlighting