fix missing import since use of __all__ was fixed