minor code tweak