diff --git a/julearn/model_selection/utils.py b/julearn/model_selection/utils.py
index 1adcd1c44..38a54352b 100644
--- a/julearn/model_selection/utils.py
+++ b/julearn/model_selection/utils.py
@@ -15,8 +15,8 @@
 
 
 def check_cv(
-    cv: CVLike, classifier: bool = False, include_final_model: bool = False
-) -> CVLike:
+    cv: "CVLike", classifier: bool = False, include_final_model: bool = False
+) -> "CVLike":
     """Check the CV instance and return the proper CV for julearn.
 
     Parameters