diff --git a/src/codeflare_sdk/templates/base-template.yaml b/src/codeflare_sdk/templates/base-template.yaml
index c2dddb6d6..92362d6d8 100644
--- a/src/codeflare_sdk/templates/base-template.yaml
+++ b/src/codeflare_sdk/templates/base-template.yaml
@@ -16,21 +16,21 @@ spec:
       custompodresources:
       - replicas: 1
         requests:
-          cpu: 2
-          memory: 8G
+          cpu: 1
+          memory: 4G
           nvidia.com/gpu: 0
         limits:
-          cpu: 2
-          memory: 8G
+          cpu: 1
+          memory: 4G
           nvidia.com/gpu: 0
       - replicas: 3
         requests:
-          cpu: 2
-          memory: 12G
+          cpu: 1
+          memory: 4G
           nvidia.com/gpu: 1
         limits:
-          cpu: 2
-          memory: 12G
+          cpu: 1
+          memory: 4G
           nvidia.com/gpu: 1
       generictemplate:
         # This config demonstrates KubeRay's Ray autoscaler integration.
diff --git a/tests/e2e/codeflare_sdk-0.0.0.dev0-py3-none-any.whl b/tests/e2e/codeflare_sdk-0.0.0.dev0-py3-none-any.whl
index 8d0cddc3b..071675dc4 100644
Binary files a/tests/e2e/codeflare_sdk-0.0.0.dev0-py3-none-any.whl and b/tests/e2e/codeflare_sdk-0.0.0.dev0-py3-none-any.whl differ
diff --git a/tests/e2e/mnist_raycluster_sdk.py b/tests/e2e/mnist_raycluster_sdk.py
index b830a0040..b5d8ac878 100644
--- a/tests/e2e/mnist_raycluster_sdk.py
+++ b/tests/e2e/mnist_raycluster_sdk.py
@@ -9,17 +9,19 @@
 
 namespace = sys.argv[1]
 
-cluster = Cluster(ClusterConfiguration(
-    name='mnist',
-    namespace=namespace,
-    num_workers=1,
-    min_cpus='500m',
-    max_cpus=1,
-    min_memory=0.5,
-    max_memory=1,
-    num_gpus=0,
-    instascale=False,
-))
+cluster = Cluster(
+    ClusterConfiguration(
+        name="mnist",
+        namespace=namespace,
+        num_workers=1,
+        min_cpus="500m",
+        max_cpus=1,
+        min_memory=0.5,
+        max_memory=1,
+        num_gpus=0,
+        instascale=False,
+    )
+)
 
 cluster.up()