another code tweak