Fix tox run