Small code tweak.