remove unneeded code from tests