Fixing testing script