Removing unneeded code in test.