[test] Updated test helper functions

I fixed the error