NameError fix