Removed unneeded code from tests