Fix installation, broken since started doing import in __init__

Thanks @myint for the catch and code suggestion