fix code highlighting for preview