Fix highlighting in vim