Fix style of tests