edited test script to make it work