Fix language highlighting