Fixes code highlighting