Fix for tox