Removed unnecessary code in tests.