Fixed testing script