diff --git a/experiments/run-benchmark.py b/experiments/run-benchmark.py
index 2c454ce..4a89e1a 100644
--- a/experiments/run-benchmark.py
+++ b/experiments/run-benchmark.py
@@ -9,7 +9,7 @@
 import platform, socket, psutil
 import argparse
 import flatnav
-from flatnav import DataType
+from flatnav.data_type import DataType
 from data_loader import get_data_loader
 from plotting.plot import create_plot, create_linestyles
 from plotting.metrics import metric_manager