diff --git a/server/src/test/java/io/spine/server/procman/ProcessManagerRepositoryTest.java b/server/src/test/java/io/spine/server/procman/ProcessManagerRepositoryTest.java
index 98abf67ab51..a570e2dfe91 100644
--- a/server/src/test/java/io/spine/server/procman/ProcessManagerRepositoryTest.java
+++ b/server/src/test/java/io/spine/server/procman/ProcessManagerRepositoryTest.java
@@ -754,8 +754,12 @@ void updateColumns() {
 
         // Check the column value is propagated to the entity state.
         TestProcessManager entityWithColumns = afterMigration.next();
-        assertThat(entityWithColumns.state()
-                                    .getIdString()).isEqualTo(id.toString());
+        Project expectedState = pm
+                .state()
+                .toBuilder()
+                .setIdString(pm.getIdString())
+                .build();
+        assertThat(entityWithColumns.state()).isEqualTo(expectedState);
     }
 
     @Test
diff --git a/server/src/test/java/io/spine/server/projection/ProjectionRepositoryTest.java b/server/src/test/java/io/spine/server/projection/ProjectionRepositoryTest.java
index aa69d2b31c8..bf8a2904655 100644
--- a/server/src/test/java/io/spine/server/projection/ProjectionRepositoryTest.java
+++ b/server/src/test/java/io/spine/server/projection/ProjectionRepositoryTest.java
@@ -657,8 +657,12 @@ void updateColumns() {
 
         // Check the column value is propagated to the entity state.
         TestProjection entityWithColumns = afterMigration.next();
-        assertThat(entityWithColumns.state()
-                                    .getIdString()).isEqualTo(id.toString());
+        Project expectedState = projection
+                .state()
+                .toBuilder()
+                .setIdString(projection.getIdString())
+                .build();
+        assertThat(entityWithColumns.state()).isEqualTo(expectedState);
     }
 
     @Test