Not happy with the tests