Fixing tox up to work