Fixed broken tests
Update version
Merge pull request #6 from marekbryling/master

Fixed broken tests (bad import)

@marekbryling Thanks for your fix.