diff --git a/src/codeocean/client.py b/src/codeocean/client.py
index 3b14056..ef15638 100644
--- a/src/codeocean/client.py
+++ b/src/codeocean/client.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from dataclasses import dataclass
 from requests_toolbelt.adapters.socket_options import TCPKeepAliveAdapter
 from requests_toolbelt.sessions import BaseUrlSession
@@ -14,7 +16,7 @@ class CodeOcean:
 
     domain: str
     token: str
-    retries: Optional[Retry] = None
+    retries: Optional[Retry | int] = 0
 
     def __post_init__(self):
         self.session = BaseUrlSession(base_url=f"{self.domain}/api/v1/")