Remove unusied import.
Try to fix the test.