From fb656d3bb1149c715f0b457004b6aa7b6e5ced1d Mon Sep 17 00:00:00 2001
From: Garrett Beatty <gcbeatty@amazon.com>
Date: Tue, 17 Feb 2026 14:57:54 -0500
Subject: [PATCH] fix test app

---
 test/AWS.Deploy.DockerImageUploader/App.cs | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/test/AWS.Deploy.DockerImageUploader/App.cs b/test/AWS.Deploy.DockerImageUploader/App.cs
index 68c2368a..ab3a220e 100644
--- a/test/AWS.Deploy.DockerImageUploader/App.cs
+++ b/test/AWS.Deploy.DockerImageUploader/App.cs
@@ -10,6 +10,7 @@
 using System.Threading.Tasks;
 using System.IO;
 using System.Linq;
+using AWS.Deploy.CLI.Extensions;
 using AWS.Deploy.CLI.Utilities;
 using AWS.Deploy.Common.Recipes;
 using AWS.Deploy.Orchestration;
@@ -64,6 +65,7 @@ private async Task CreateImageAndPushToECR(string projectPath)
             var configFilePath = Path.Combine(projectPath, "DockerImageUploaderConfigFile.json");
             var deployArgs = new[] { "deploy", "--project-path", projectPath, "--diagnostics", "--apply", configFilePath, "--silent" };
             var serviceCollection = new ServiceCollection();
+            serviceCollection.AddCustomServices();
 
             var registrar = new TypeRegistrar(serviceCollection);