Minor code tweak.