Fixing test script