diff --git a/app/MindWork AI Studio/Tools/RAG/RetrievalImageContext.cs b/app/MindWork AI Studio/Tools/RAG/RetrievalImageContext.cs
index 37cadb98..973892d3 100644
--- a/app/MindWork AI Studio/Tools/RAG/RetrievalImageContext.cs	
+++ b/app/MindWork AI Studio/Tools/RAG/RetrievalImageContext.cs	
@@ -2,7 +2,7 @@ using AIStudio.Chat;
 
 namespace AIStudio.Tools.RAG;
 
-public sealed class RetrievalImageContext : IRetrievalContext
+public sealed class RetrievalImageContext : IRetrievalContext, IImageSource
 {
     #region Implementation of IRetrievalContext