diff --git a/client/datalake/queue.py b/client/datalake/queue.py
index 5f47cf4..6b24bdc 100644
--- a/client/datalake/queue.py
+++ b/client/datalake/queue.py
@@ -222,7 +222,7 @@ def _run(self, timeout):
 
     def _update_time_remaining(self):
         if self._run_time_remaining is self.INFINITY:
-            return self.INFINITY
+            return float(inf)
         now = time.time()
         duration = now - self._run_start
         self._run_time_remaining -= duration