Fixed problem with test script