Fix test script to fits new cli