Fix __init__.py argument nb