Making the tests smaller