Fixing error when define __all__