Fixes tests on tox