Fix for issue #7 (`ImportError: cannot import name hmm`)