removed unnecessary import in tests