diff --git a/src/chocolatey/infrastructure.app/services/INugetService.cs b/src/chocolatey/infrastructure.app/services/INugetService.cs
index 22e9debea..77083f181 100644
--- a/src/chocolatey/infrastructure.app/services/INugetService.cs
+++ b/src/chocolatey/infrastructure.app/services/INugetService.cs
@@ -19,6 +19,7 @@ namespace chocolatey.infrastructure.app.services
     using System.Collections.Concurrent;
     using System.Collections.Generic;
     using configuration;
+    using domain;
     using results;
 
     public interface INugetService : ISourceRunner
@@ -65,5 +66,13 @@ public interface INugetService : ISourceRunner
         /// </summary>
         /// <param name="config">The configuration</param>
         IEnumerable<PackageResult> get_all_installed_packages(ChocolateyConfiguration config);
+
+        /// <summary>
+        /// Sets the configuration for the package upgrade
+        /// </summary>
+        /// <param name="config">The configuration.</param>
+        /// <param name="packageInfo">The package information.</param>
+        /// <returns>The original unmodified configuration, so it can be reset after upgrade</returns>
+        ChocolateyConfiguration set_package_config_for_upgrade(ChocolateyConfiguration config, ChocolateyPackageInformation packageInfo);
     }
 }
diff --git a/src/chocolatey/infrastructure.app/services/NugetService.cs b/src/chocolatey/infrastructure.app/services/NugetService.cs
index 73ea509a2..47ec56866 100644
--- a/src/chocolatey/infrastructure.app/services/NugetService.cs
+++ b/src/chocolatey/infrastructure.app/services/NugetService.cs
@@ -1453,7 +1453,7 @@ public virtual ConcurrentDictionary<string, PackageResult> get_outdated(Chocolat
         /// <param name="config">The configuration.</param>
         /// <param name="packageInfo">The package information.</param>
         /// <returns>The original unmodified configuration, so it can be reset after upgrade</returns>
-        protected virtual ChocolateyConfiguration set_package_config_for_upgrade(ChocolateyConfiguration config, ChocolateyPackageInformation packageInfo)
+        public virtual ChocolateyConfiguration set_package_config_for_upgrade(ChocolateyConfiguration config, ChocolateyPackageInformation packageInfo)
         {
             if (!config.Features.UseRememberedArgumentsForUpgrades || string.IsNullOrWhiteSpace(packageInfo.Arguments)) return config;