refactored the tests a little to make them more readable