diff --git a/nebari_workflow_controller/app.py b/nebari_workflow_controller/app.py
index 66ab241..ab07d2e 100644
--- a/nebari_workflow_controller/app.py
+++ b/nebari_workflow_controller/app.py
@@ -309,7 +309,6 @@ def mutate_template(container_keep_portions, spec_keep_portions, template):
                     **value,
                 }
             else:
-                # TODO: I think this will throw error if template doesn't have container key
                 template["container"][key] = value
         elif isinstance(value, list):
             if key in template["container"]:
diff --git a/pyproject.toml b/pyproject.toml
index 50d4320..de36fdc 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -51,6 +51,7 @@ dependencies = [
     "uvicorn==0.21.1",
     "pydantic==1.10.7",
     "jsonpatch==1.32",
+    "kubernetes==26.1.0",
 ]
 
 [project.optional-dependencies]