small code tweak