diff --git a/src/batdetect2/cli/evaluate.py b/src/batdetect2/cli/evaluate.py
index 6b1afe5..4da81dd 100644
--- a/src/batdetect2/cli/evaluate.py
+++ b/src/batdetect2/cli/evaluate.py
@@ -37,7 +37,7 @@ def evaluate_command(
     run_name: Optional[str] = None,
     verbose: int = 0,
 ):
-    from batdetect2.api.base import BatDetect2API
+    from batdetect2.api_v2 import BatDetect2API
     from batdetect2.config import load_full_config
     from batdetect2.data import load_dataset_from_config
 
diff --git a/src/batdetect2/cli/train.py b/src/batdetect2/cli/train.py
index 8e9f451..f0168a3 100644
--- a/src/batdetect2/cli/train.py
+++ b/src/batdetect2/cli/train.py
@@ -46,7 +46,7 @@ def train_command(
     run_name: Optional[str] = None,
     verbose: int = 0,
 ):
-    from batdetect2.api.base import BatDetect2API
+    from batdetect2.api_v2 import BatDetect2API
     from batdetect2.config import (
         BatDetect2Config,
         load_full_config,