Trying to make the tests work