Update project to run correctly with tox