Fix source code highlight