Fixing tests
Hopefully this script works