Simplify test code (#679)

* Simplify test code

* fix
1 file changed