Fix import path, fixes #9