Fixed import error
Merge pull request #177 from nwinklareth/v2.0-dev

Fixed import error