diff --git a/jps/jps-plugin/jps-tests/test/org/jetbrains/kotlin/jps/incremental/AbstractProtoComparisonTest.kt b/jps/jps-plugin/jps-tests/test/org/jetbrains/kotlin/jps/incremental/AbstractProtoComparisonTest.kt
index 9e1c15e1268..4065354f67c 100644
--- a/jps/jps-plugin/jps-tests/test/org/jetbrains/kotlin/jps/incremental/AbstractProtoComparisonTest.kt
+++ b/jps/jps-plugin/jps-tests/test/org/jetbrains/kotlin/jps/incremental/AbstractProtoComparisonTest.kt
@@ -58,10 +58,20 @@ abstract class AbstractProtoComparisonTest<PROTO_DATA> : TestWithWorkingDir() {
 
             val rawProtoDifference = when {
                 oldData is ClassProtoData && newData is ClassProtoData -> {
-                    ProtoCompareGenerated(oldData.nameResolver, newData.nameResolver).difference(oldData.proto, newData.proto)
+                    ProtoCompareGenerated(
+                        oldNameResolver = oldData.nameResolver,
+                        newNameResolver = newData.nameResolver,
+                        oldTypeTable = oldData.proto.typeTableOrNull,
+                        newTypeTable = newData.proto.typeTableOrNull
+                    ).difference(oldData.proto, newData.proto)
                 }
                 oldData is PackagePartProtoData && newData is PackagePartProtoData -> {
-                    ProtoCompareGenerated(oldData.nameResolver, newData.nameResolver).difference(oldData.proto, newData.proto)
+                    ProtoCompareGenerated(
+                        oldNameResolver = oldData.nameResolver,
+                        newNameResolver = newData.nameResolver,
+                        oldTypeTable = oldData.proto.typeTableOrNull,
+                        newTypeTable = newData.proto.typeTableOrNull
+                    ).difference(oldData.proto, newData.proto)
                 }
                 else -> null
             }