Fixing NameError described in issue #1

:)