diff --git a/Disco.Plugins.ADCompare.csproj b/Disco.Plugins.ADCompare.csproj
index 805c5df..9fcc9cb 100644
--- a/Disco.Plugins.ADCompare.csproj
+++ b/Disco.Plugins.ADCompare.csproj
@@ -68,6 +68,8 @@
     <Compile Include="ADComparePlugin.cs" />
     <Compile Include="ConfigurationHandler\ADCompareConfigurationHandler.cs" />
     <Compile Include="Features\DeviceCompareService.cs" />
+    <Compile Include="Features\DeviceUIExtension.cs" />
+    <Compile Include="Features\UserUIExtension.cs" />
     <Compile Include="Models\DeviceComparisonResult.cs" />
     <Compile Include="WebHandler\ADCompareWebHandler.cs" />
     <Compile Include="Properties\AssemblyInfo.cs" />