Removed unnecessary code from tests.