Fixed the tests and now it passes tox!