Removed unnecessary import in test.