Remove code from the tests