I think this ought to be an absolute import for Python 3. If it's having issues, can add `from __future__ import absolute_imports` to fix it.