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
///
/// The configuration
IEnumerable get_all_installed_packages(ChocolateyConfiguration config);
+
+ ///
+ /// Sets the configuration for the package upgrade
+ ///
+ /// The configuration.
+ /// The package information.
+ /// The original unmodified configuration, so it can be reset after upgrade
+ 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 get_outdated(Chocolat
/// The configuration.
/// The package information.
/// The original unmodified configuration, so it can be reset after upgrade
- 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;