Refactor the tests a little bit