diff --git a/benchs/bench_big_batch_ivf.py b/benchs/bench_big_batch_ivf.py
index 74ca10c02c..e678d8e7a2 100644
--- a/benchs/bench_big_batch_ivf.py
+++ b/benchs/bench_big_batch_ivf.py
@@ -11,7 +11,7 @@
 import numpy as np
 
 from faiss.contrib.datasets import SyntheticDataset
-from faiss.contrib.ivf_tools import big_batch_search
+from faiss.contrib.big_batch_search import big_batch_search
 
 parser = argparse.ArgumentParser()