diff --git a/packages/SystemUI/src/com/android/systemui/assist/AssistLogger.kt b/packages/SystemUI/src/com/android/systemui/assist/AssistLogger.kt
index e651392756856..08edad3a7f7fa 100644
--- a/packages/SystemUI/src/com/android/systemui/assist/AssistLogger.kt
+++ b/packages/SystemUI/src/com/android/systemui/assist/AssistLogger.kt
@@ -63,7 +63,7 @@ open class AssistLogger @Inject constructor(
     }
 
     fun reportAssistantInvocationEvent(
-        invocationEvent: AssistantInvocationEvent,
+        invocationEvent: UiEventLogger.UiEventEnum,
         assistantComponent: ComponentName? = null,
         deviceState: Int? = null
     ) {
@@ -86,7 +86,7 @@ open class AssistLogger @Inject constructor(
         reportAssistantInvocationExtraData()
     }
 
-    fun reportAssistantSessionEvent(sessionEvent: AssistantSessionEvent) {
+    fun reportAssistantSessionEvent(sessionEvent: UiEventLogger.UiEventEnum) {
         val assistantComponent = getAssistantComponentForCurrentUser()
         val assistantUid = getAssistantUid(assistantComponent)
         uiEventLogger.logWithInstanceId(