Fixing NameError. My bad.