diff --git a/python/openshell/sandbox.py b/python/openshell/sandbox.py
index 7dbf1b9d4..520e00a5a 100644
--- a/python/openshell/sandbox.py
+++ b/python/openshell/sandbox.py
@@ -1736,9 +1736,9 @@ def _sandbox_workload_template(
     if environment:
         template.spec.workload.environment.update(dict(environment))
     if cpu is not None:
-        template.spec.workload.resources.cpu = cpu
+        template.spec.workload.resources.cpu.quantity = cpu
     if memory is not None:
-        template.spec.workload.resources.memory = memory
+        template.spec.workload.resources.memory.quantity = memory
     if gpu or gpu_count is not None:
         template.spec.workload.resources.gpu.SetInParent()
     if gpu_count is not None:
diff --git a/python/openshell/sandbox_test.py b/python/openshell/sandbox_test.py
index 4b8e60e07..4bad39f83 100644
--- a/python/openshell/sandbox_test.py
+++ b/python/openshell/sandbox_test.py
@@ -1960,8 +1960,8 @@ def _make_workload_template_proto(
     template.metadata.name = name
     template.metadata.workspace = workspace
     template.spec.workload.image = f"ghcr.io/test/{name}:latest"
-    template.spec.workload.resources.cpu = "1"
-    template.spec.workload.resources.memory = "512Mi"
+    template.spec.workload.resources.cpu.quantity = "1"
+    template.spec.workload.resources.memory.quantity = "512Mi"
     return template
 
 
@@ -2256,8 +2256,8 @@ def test_sandbox_template_create_builds_template_from_public_fields() -> None:
     assert dict(template.metadata.annotations) == {"owner": "platform"}
     assert template.spec.workload.image == "ghcr.io/test/gpu-kata:latest"
     assert dict(template.spec.workload.environment) == {"FEATURE_FLAG": "on"}
-    assert template.spec.workload.resources.cpu == "1"
-    assert template.spec.workload.resources.memory == "512Mi"
+    assert template.spec.workload.resources.cpu.quantity == "1"
+    assert template.spec.workload.resources.memory.quantity == "512Mi"
     assert template.spec.workload.resources.gpu.count == 2
     assert template.spec.driver_config["kubernetes"]["runtime_class_name"] == "kata"