diff --git a/benchmarks/domains/__init__.py b/benchmarks/domains/__init__.py
index 9e761cfc1..2463687e1 100644
--- a/benchmarks/domains/__init__.py
+++ b/benchmarks/domains/__init__.py
@@ -1,6 +1,6 @@
 """Benchmark domains."""
 
-from benchmarks.definition.config import Benchmark
+from benchmarks.definition.base import Benchmark
 from benchmarks.domains.synthetic_2C1D_1C import synthetic_2C1D_1C_benchmark
 from benchmarks.domains.Hardness import hardness_benchmark, hardness_transfer_learning_benchmark