Updated test script to prevent error.