diff --git a/src/renderer/components/Experiment/Rag/Documents.tsx b/src/renderer/components/Experiment/Rag/Documents.tsx
index de27b787de11167c5610d7754eab406c1c483947..142c7402bd16fdc43a6e246ab749e798a67b96d9 100644
--- a/src/renderer/components/Experiment/Rag/Documents.tsx
+++ b/src/renderer/components/Experiment/Rag/Documents.tsx
@@ -140,6 +140,7 @@ export default function Documents({
   experimentInfo,
   fullPage = false,
   additionalMessage = false,
+  fixedFolder = '',
 }) {
   const [doc, setDoc] = React.useState<Doc>('desc');
   const [selected, setSelected] = React.useState<readonly string[]>([]);
@@ -154,13 +155,7 @@ export default function Documents({
 
   const [loading, setLoading] = React.useState(false);
 
-  const [currentFolder, setCurrentFolder] = React.useState('');
-
-  React.useEffect(() => {
-    if (!additionalMessage) {
-      setCurrentFolder('rag');
-    }
-  }, [additionalMessage]);
+  const [currentFolder, setCurrentFolder] = React.useState(fixedFolder);
 
   const {
     data: rows,
@@ -474,10 +469,9 @@ export default function Documents({
           ) : (
             <>
               <Link
+                disabled={fixedFolder !== ''}
                 onClick={() => {
-                  additionalMessage
-                    ? setCurrentFolder('')
-                    : setCurrentFolder('rag');
+                  setCurrentFolder('');
                 }}
               >
                 Documents /
diff --git a/src/renderer/components/Experiment/Rag/index.tsx b/src/renderer/components/Experiment/Rag/index.tsx
index e008b8d7c426fbe64493ac1e319a0b69c0b4ee44..7a754ee38daa1ab9d2e71f6ceca70e9453e5dec9 100644
--- a/src/renderer/components/Experiment/Rag/index.tsx
+++ b/src/renderer/components/Experiment/Rag/index.tsx
@@ -117,7 +117,7 @@ export default function DocumentSearch({ experimentInfo, setRagEngine }) {
             flexDirection: 'column',
           }}
         >
-          <Documents experimentInfo={experimentInfo} />
+          <Documents experimentInfo={experimentInfo} setFolder="rag" />
         </Box>
         <Box sx={{ flex: 1, display: 'flex', flexDirection: 'column' }}>
           <Query experimentInfo={experimentInfo} />