fix __all__, use the correct argument name