Removed unnecessary code from the tests