Fix code highlighting