Removed unneccessary code in test.