improve style of the tests file
0.4.0