diff --git a/src/DesignTimeStyleHelper/App.xaml.cs b/src/DesignTimeStyleHelper/App.xaml.cs
index 265c1e7f1d..6fba7ef271 100644
--- a/src/DesignTimeStyleHelper/App.xaml.cs
+++ b/src/DesignTimeStyleHelper/App.xaml.cs
@@ -11,6 +11,7 @@
 using Microsoft.VisualStudio.ComponentModelHost;
 using Microsoft.VisualStudio.Shell;
 using Moq;
+using GitHub.Models;
 
 namespace DesignTimeStyleHelper
 {
@@ -42,7 +43,7 @@ public class Holder
         public IVisualStudioBrowser Browser;
 
         [Import]
-        public ExportFactoryProvider ExportFactoryProvider;
+        public IExportFactoryProvider ExportFactoryProvider;
 
         [Import]
         public IUIProvider UIProvider;
@@ -68,7 +69,7 @@ public CustomServiceProvider()
         {
             catalog = new AggregateCatalog(
                 new AssemblyCatalog(typeof(CustomServiceProvider).Assembly),
-                new AssemblyCatalog(typeof(Services).Assembly), // GitHub.VisualStudio
+                new AssemblyCatalog(typeof(Program).Assembly), // GitHub.VisualStudio
                 new AssemblyCatalog(typeof(GitHub.Api.ApiClient).Assembly), // GitHub.App
                 new AssemblyCatalog(typeof(GitHub.Api.SimpleApiClient).Assembly), // GitHub.Api
                 new AssemblyCatalog(typeof(Rothko.Environment).Assembly), // Rothko
diff --git a/src/DesignTimeStyleHelper/DesignTimeStyleHelper.csproj b/src/DesignTimeStyleHelper/DesignTimeStyleHelper.csproj
index c7c745bfc0..f16a881b5c 100644
--- a/src/DesignTimeStyleHelper/DesignTimeStyleHelper.csproj
+++ b/src/DesignTimeStyleHelper/DesignTimeStyleHelper.csproj
@@ -75,6 +75,10 @@
     <Reference Include="WindowsBase" />
     <Reference Include="PresentationCore" />
     <Reference Include="PresentationFramework" />
+    <Reference Include="Microsoft.TeamFoundation.Git.Provider, Version=14.0.0.0, Culture=neutral, PublicKeyToken=b03f5f7f11d50a3a, processorArchitecture=MSIL">
+      <HintPath>..\..\lib\Microsoft.TeamFoundation.Git.Provider.dll</HintPath>
+      <Private>True</Private>
+    </Reference>
   </ItemGroup>
   <ItemGroup>
     <ApplicationDefinition Include="App.xaml">