Fix import error

This PR encompasses #68.