minor code tweak.