diff --git a/py/kubeflow/testing/cleanup_ci.py b/py/kubeflow/testing/cleanup_ci.py
index 8ae549d5a..8ffd519dc 100644
--- a/py/kubeflow/testing/cleanup_ci.py
+++ b/py/kubeflow/testing/cleanup_ci.py
@@ -1161,7 +1161,7 @@ def main():
   args = parser.parse_args()
 
   # Update max age
-  MAX_LIFETIME[E2E_INFRA] = args.max_age_hours
+  MAX_LIFETIME[E2E_INFRA] = datetime.timedelta(hours=args.max_age_hours)
 
   logging.info("Max lifetime:\n%s", MAX_LIFETIME)