diff --git a/OnnxStack.UI/Views/ImageInpaint.xaml.cs b/OnnxStack.UI/Views/ImageInpaint.xaml.cs
index a10a05cb..9f212969 100644
--- a/OnnxStack.UI/Views/ImageInpaint.xaml.cs
+++ b/OnnxStack.UI/Views/ImageInpaint.xaml.cs
@@ -25,7 +25,7 @@ namespace OnnxStack.UI.Views
     /// </summary>
     public partial class ImageInpaint : UserControl, INavigatable, INotifyPropertyChanged
     {
-        private readonly ILogger<TextToImageView> _logger;
+        private readonly ILogger<ImageInpaint> _logger;
         private readonly IStableDiffusionService _stableDiffusionService;
 
         private bool _hasResult;
@@ -52,7 +52,7 @@ public ImageInpaint()
         {
             if (!DesignerProperties.GetIsInDesignMode(this))
             {
-                _logger = App.GetService<ILogger<TextToImageView>>();
+                _logger = App.GetService<ILogger<ImageInpaint>>();
                 _stableDiffusionService = App.GetService<IStableDiffusionService>();
             }
 
diff --git a/OnnxStack.UI/Views/ImageToImage.xaml.cs b/OnnxStack.UI/Views/ImageToImage.xaml.cs
index 68b9f664..e6a2c020 100644
--- a/OnnxStack.UI/Views/ImageToImage.xaml.cs
+++ b/OnnxStack.UI/Views/ImageToImage.xaml.cs
@@ -26,7 +26,7 @@ namespace OnnxStack.UI.Views
     /// </summary>
     public partial class ImageToImage : UserControl, INavigatable, INotifyPropertyChanged
     {
-        private readonly ILogger<TextToImageView> _logger;
+        private readonly ILogger<ImageToImage> _logger;
         private readonly IStableDiffusionService _stableDiffusionService;
 
         private bool _hasResult;
@@ -51,7 +51,7 @@ public ImageToImage()
         {
             if (!DesignerProperties.GetIsInDesignMode(this))
             {
-                _logger = App.GetService<ILogger<TextToImageView>>();
+                _logger = App.GetService<ILogger<ImageToImage>>();
                 _stableDiffusionService = App.GetService<IStableDiffusionService>();
             }