Remove import of mock.