Fixing bug in test script.