diff --git a/app/server/appsmith-server/src/main/java/com/appsmith/server/configurations/ce/CustomCookieWebSessionIdResolverCE.java b/app/server/appsmith-server/src/main/java/com/appsmith/server/configurations/ce/CustomCookieWebSessionIdResolverCE.java
index 481a94d50bb4..181661507a06 100644
--- a/app/server/appsmith-server/src/main/java/com/appsmith/server/configurations/ce/CustomCookieWebSessionIdResolverCE.java
+++ b/app/server/appsmith-server/src/main/java/com/appsmith/server/configurations/ce/CustomCookieWebSessionIdResolverCE.java
@@ -27,11 +27,11 @@ public CustomCookieWebSessionIdResolverCE() {
 
     @Override
     public void setSessionId(ServerWebExchange exchange, String id) {
-        addCookieInitializers();
+        addCookieInitializers(exchange);
         super.setSessionId(exchange, id);
     }
 
-    protected void addCookieInitializers() {
+    protected void addCookieInitializers(ServerWebExchange exchange) {
         // Add the appropriate SameSite attribute based on the exchange attribute
         addCookieInitializer((builder) -> builder.sameSite(LAX).secure(true));
     }