diff --git a/ci/notebook_list.py b/ci/notebook_list.py
index 55fdd33735..db26f2efa2 100644
--- a/ci/notebook_list.py
+++ b/ci/notebook_list.py
@@ -17,6 +17,8 @@
 import glob
 from pathlib import Path
 
+from cuda.bindings import runtime
+
 # for adding another run type and skip file name add to this dictionary
 runtype_dict = {
     "all": "",