Fixed testing script to fit the new integration testing changes.