Fixed testing script.