fix the step highlight code