Fix ImportError for Python 3

Refs #5