Drop unneeded code in unit tests