Fix shell highlight