diff --git a/java/dagger/hilt/android/plugin/main/src/main/kotlin/dagger/hilt/android/plugin/util/Tasks.kt b/java/dagger/hilt/android/plugin/main/src/main/kotlin/dagger/hilt/android/plugin/util/Tasks.kt
index 29c116a17db..a91ce35c6e1 100644
--- a/java/dagger/hilt/android/plugin/main/src/main/kotlin/dagger/hilt/android/plugin/util/Tasks.kt
+++ b/java/dagger/hilt/android/plugin/main/src/main/kotlin/dagger/hilt/android/plugin/util/Tasks.kt
@@ -49,7 +49,9 @@ internal fun addKaptTaskProcessorOptions(
     """.trimIndent()
   }
   project.tasks.withType(KaptTask::class.java) { task ->
-    if (task.name == "kapt${component.name.capitalize()}Kotlin") {
+    if (task.name == "kapt${component.name.capitalize()}Kotlin" ||
+        // Task names in shared/src/AndroidMain in KMP projects has a platform suffix.
+        task.name == "kapt${component.name.capitalize()}KotlinAndroid") {
       val argProvider = produceArgProvider.invoke(task)
       // TODO: Update once KT-58009 is fixed.
       try {
@@ -83,7 +85,9 @@ internal fun addKspTaskProcessorOptions(
     """.trimIndent()
   }
   project.tasks.withType(KspTaskJvm::class.java) { task ->
-    if (task.name == "ksp${component.name.capitalize()}Kotlin") {
+    if (task.name == "ksp${component.name.capitalize()}Kotlin" ||
+        // Task names in shared/src/AndroidMain in KMP projects has a platform suffix.
+        task.name == "ksp${component.name.capitalize()}KotlinAndroid") {
       task.commandLineArgumentProviders.add(produceArgProvider.invoke(task))
     }
   }
@@ -102,4 +106,5 @@ internal fun Task.isKspTask(): Boolean = try {
   kspTaskClass.isAssignableFrom(this::class.java)
 } catch (ex: ClassNotFoundException) {
   false
-}
\ No newline at end of file
+}
+