removed unnecessary code in tests