diff --git a/common-util/src/main/kotlin/com/google/devtools/ksp/common/IncrementalContextBase.kt b/common-util/src/main/kotlin/com/google/devtools/ksp/common/IncrementalContextBase.kt
index 1bb3add83f..148d4324db 100644
--- a/common-util/src/main/kotlin/com/google/devtools/ksp/common/IncrementalContextBase.kt
+++ b/common-util/src/main/kotlin/com/google/devtools/ksp/common/IncrementalContextBase.kt
@@ -234,9 +234,7 @@ abstract class IncrementalContextBase(
             symbolLookupCache.get(it).map { File(it) }
         }
 
-        val dirtyFilesBySealed = sealedMap.keys.flatMap { sealedMap[it]!! }.flatMap {
-            symbolLookupCache.get(it).map { File(it) }
-        }
+        val dirtyFilesBySealed = sealedMap.keys
 
         // Calculate dirty files by dirty classes in CP.
         val dirtyFilesByCP = changedClasses.flatMap { fqn ->
diff --git a/integration-tests/src/test/kotlin/com/google/devtools/ksp/test/GetSealedSubclassesIncIT.kt b/integration-tests/src/test/kotlin/com/google/devtools/ksp/test/GetSealedSubclassesIncIT.kt
index d9cfa04dba..3610aba833 100644
--- a/integration-tests/src/test/kotlin/com/google/devtools/ksp/test/GetSealedSubclassesIncIT.kt
+++ b/integration-tests/src/test/kotlin/com/google/devtools/ksp/test/GetSealedSubclassesIncIT.kt
@@ -2,7 +2,6 @@ package com.google.devtools.ksp.test
 
 import org.gradle.testkit.runner.GradleRunner
 import org.junit.Assert
-import org.junit.Assume
 import org.junit.Rule
 import org.junit.Test
 import org.junit.runner.RunWith
@@ -17,7 +16,6 @@ class GetSealedSubclassesIncIT(val useKSP2: Boolean) {
 
     @Test
     fun testGetSealedSubclassesInc() {
-        Assume.assumeFalse(useKSP2)
         val gradleRunner = GradleRunner.create().withProjectDir(project.root)
 
         val expected2 = listOf(