diff --git a/VisualStudioDiscordRPC.Shared/Variables/EditionVariable.cs b/VisualStudioDiscordRPC.Shared/Variables/EditionVariable.cs
index c9cc171..694c13a 100644
--- a/VisualStudioDiscordRPC.Shared/Variables/EditionVariable.cs
+++ b/VisualStudioDiscordRPC.Shared/Variables/EditionVariable.cs
@@ -1,4 +1,4 @@
-using EnvDTE;
+using EnvDTE80;
 
 namespace VisualStudioDiscordRPC.Shared.Variables
 {
@@ -6,7 +6,7 @@ public class EditionVariable : Variable
     {
         private readonly string _edition;
 
-        public EditionVariable(DTE dte)
+        public EditionVariable(DTE2 dte)
         {
             Microsoft.VisualStudio.Shell.ThreadHelper.ThrowIfNotOnUIThread();
             _edition = dte.Edition;
diff --git a/VisualStudioDiscordRPC.Shared/Variables/VersionVariable.cs b/VisualStudioDiscordRPC.Shared/Variables/VersionVariable.cs
index 2f47def..1dc56c1 100644
--- a/VisualStudioDiscordRPC.Shared/Variables/VersionVariable.cs
+++ b/VisualStudioDiscordRPC.Shared/Variables/VersionVariable.cs
@@ -1,4 +1,4 @@
-using EnvDTE;
+using EnvDTE80;
 using VisualStudioDiscordRPC.Shared.Utils;
 
 namespace VisualStudioDiscordRPC.Shared.Variables
@@ -7,7 +7,7 @@ public class VersionVariable : Variable
     {
         private readonly string _version;
 
-        public VersionVariable(DTE dte) 
+        public VersionVariable(DTE2 dte) 
         {
             Microsoft.VisualStudio.Shell.ThreadHelper.ThrowIfNotOnUIThread();
             _version = dte.Version;