Removed unneeded imports and useless code from the test file.