Fix language highlight