Fix js highlighting