Revert change to __all__