fixing NameError in previous commit