diff --git a/internal/states/remote/state_test.go b/internal/states/remote/state_test.go
index cbb6a6219f07..8d207f7a355e 100644
--- a/internal/states/remote/state_test.go
+++ b/internal/states/remote/state_test.go
@@ -129,8 +129,9 @@ func TestStatePersist(t *testing.T) {
 										"attributes_flat": map[string]interface{}{
 											"filename": "file.txt",
 										},
-										"schema_version":       0.0,
-										"sensitive_attributes": []interface{}{},
+										"identity_schema_version": 0.0,
+										"schema_version":          0.0,
+										"sensitive_attributes":    []interface{}{},
 									},
 								},
 								"mode":     "managed",
@@ -167,8 +168,9 @@ func TestStatePersist(t *testing.T) {
 										"attributes_flat": map[string]interface{}{
 											"filename": "file.txt",
 										},
-										"schema_version":       0.0,
-										"sensitive_attributes": []interface{}{},
+										"identity_schema_version": 0.0,
+										"schema_version":          0.0,
+										"sensitive_attributes":    []interface{}{},
 									},
 								},
 								"mode":     "managed",