Small code tweak to make it more understandable