diff --git a/dockerfiles/non-shm.yml b/dockerfiles/non-shm.yml
index 95256d1..33ce294 100644
--- a/dockerfiles/non-shm.yml
+++ b/dockerfiles/non-shm.yml
@@ -6,11 +6,13 @@ spec:
   containers:
   - name: server
     image: cheneyyu/shm-demo
-    env:
-      - name: ROLE
-        value: s
+    resources:
+      limits:
+        cpu: "1"
+        memory: "1G"
   - name: function-1
     image: cheneyyu/shm-demo
-    env:
-      - name: ROLE
-        value: c
\ No newline at end of file
+    resources:
+      limits:
+        cpu: "1"
+        memory: "1G"
\ No newline at end of file
diff --git a/dockerfiles/shm.yml b/dockerfiles/shm.yml
index c2dec6f..5da5711 100644
--- a/dockerfiles/shm.yml
+++ b/dockerfiles/shm.yml
@@ -6,18 +6,20 @@ spec:
   containers:
   - name: server
     image: cheneyyu/shm-demo
-    env:
-      - name: ROLE
-        value: s
+    resources:
+      limits:
+        cpu: "1"
+        memory: "1G"
     volumeMounts:
     - name: cache-volume
       mountPath: /dev/shm
 
   - name: function-1
     image: cheneyyu/shm-demo
-    env:
-      - name: ROLE
-        value: c
+    resources:
+      limits:
+        cpu: "1"
+        memory: "1G"
     volumeMounts:
     - name: cache-volume
       mountPath: /dev/shm