Update __init__.py

Fixed axes typo.