Fix issue where code highlighting was not working