Fix __all__ bug