Remove unneeded code in tests.