BUGFIX: fix NameError