Make it work for test script v4