small code tweak.