Update __init__.py

Fix for #8

TODO: Should probably be handled in ptools really.