diff --git a/Packages/com.unity.inputsystem/InputSystem/AssemblyInfo.cs b/Packages/com.unity.inputsystem/InputSystem/AssemblyInfo.cs
index 96a9c404f1..984a742195 100644
--- a/Packages/com.unity.inputsystem/InputSystem/AssemblyInfo.cs
+++ b/Packages/com.unity.inputsystem/InputSystem/AssemblyInfo.cs
@@ -8,6 +8,7 @@
 [assembly: InternalsVisibleTo("Unity.InputSystem.Tests")]
 [assembly: InternalsVisibleTo("Unity.InputSystem.IntegrationTests")]
 [assembly: InternalsVisibleTo("Unity.InputSystem.ForUI")] // To avoid minor bump
+[assembly: InternalsVisibleTo("Unity.AI.Assistant.Editor")]
 
 namespace UnityEngine.InputSystem
 {