diff --git a/e2e/setup/setup_test.go b/e2e/setup/setup_test.go
index bb15612d..0f545cc8 100644
--- a/e2e/setup/setup_test.go
+++ b/e2e/setup/setup_test.go
@@ -168,11 +168,11 @@ func Test_ConfigMap(t *testing.T) {
 
 		cmMap := cmToMap(cm.Data["nats.conf"])
 
-		if err := checkValueInCMMap(cmMap, "max_file", FileStorageSize); err != nil {
+		if err := checkValueInCMMap(cmMap, "max_file_store", FileStorageSize); err != nil {
 			return err
 		}
 
-		if err := checkValueInCMMap(cmMap, "max_mem", MemStorageSize); err != nil {
+		if err := checkValueInCMMap(cmMap, "max_memory_store", MemStorageSize); err != nil {
 			return err
 		}
 
diff --git a/testutils/integration/integration.go b/testutils/integration/integration.go
index 02cfe25f..ddb761f1 100644
--- a/testutils/integration/integration.go
+++ b/testutils/integration/integration.go
@@ -485,7 +485,7 @@ func (env TestEnvironment) EnsureNATSSpecFileStorageReflected(t *testing.T, nats
 		}
 
 		// check if file storage size is correctly defined in NATS config.
-		if !strings.Contains(natsConfigStr, fmt.Sprintf("max_file: %s", nats.Spec.FileStorage.Size.String())) {
+		if !strings.Contains(natsConfigStr, fmt.Sprintf("max_file_store: %s", nats.Spec.FileStorage.Size.String())) {
 			return false
 		}
 
@@ -529,7 +529,7 @@ func (env TestEnvironment) EnsureNATSSpecMemStorageReflected(t *testing.T, nats
 		}
 
 		// check if mem storage size is correctly defined in NATS config.
-		return strings.Contains(natsConfigStr, fmt.Sprintf("max_mem: %s", nats.Spec.MemStorage.Size.String()))
+		return strings.Contains(natsConfigStr, fmt.Sprintf("max_memory_store: %s", nats.Spec.MemStorage.Size.String()))
 	}, SmallTimeOut, SmallPollingInterval, "failed to ensure NATS CR Spec.jetStream.memStorage")
 }