diff --git a/src/python/python_py_ext.pyi b/src/python/python_py_ext.pyi
index 3003107..b643348 100644
--- a/src/python/python_py_ext.pyi
+++ b/src/python/python_py_ext.pyi
@@ -8,4 +8,4 @@
 
 from typing import Optional
 
-def get_gpu_node_id(cache_file_path: Optional[str]) -> str: ...
+def get_gpu_node_id(cache_file_path: Optional[str] = None) -> str: ...