Fix NameError
Fix the tests require entry in setup.