Fix code highlighting bug