diff --git a/project/Aki.Custom/Patches/CustomAiPatch.cs b/project/Aki.Custom/Patches/CustomAiPatch.cs
index 2a3456a..228bbcd 100644
--- a/project/Aki.Custom/Patches/CustomAiPatch.cs
+++ b/project/Aki.Custom/Patches/CustomAiPatch.cs
@@ -3,6 +3,7 @@ using Aki.Reflection.Patching;
 using EFT;
 using Newtonsoft.Json;
 using System;
+using Comfort.Common;
 using System.Collections.Generic;
 using System.Linq;
 using System.Reflection;