Fixed some errors on the test script.

https://app.asana.com/0/19128794071958/19599640169229