Fix deprecated imports
Fix #17 Lots of error logging when running tests
0.4.1