diff --git a/elasticsearch/src/test/java/org/elasticsearch/xpack/ml/job/persistence/MockBatchedDocumentsIterator.java b/elasticsearch/src/test/java/org/elasticsearch/xpack/ml/job/persistence/MockBatchedDocumentsIterator.java
index 06a0ce676ac..f2eccaa9db1 100644
--- a/elasticsearch/src/test/java/org/elasticsearch/xpack/ml/job/persistence/MockBatchedDocumentsIterator.java
+++ b/elasticsearch/src/test/java/org/elasticsearch/xpack/ml/job/persistence/MockBatchedDocumentsIterator.java
@@ -12,7 +12,7 @@ import java.util.Deque;
 import java.util.List;
 import java.util.NoSuchElementException;
 
-import static org.elasticsearch.mock.orig.Mockito.mock;
+import static org.mockito.Mockito.mock;
 
 public class MockBatchedDocumentsIterator<T> extends BatchedDocumentsIterator<T> {
     private final List<Deque<T>> batches;