diff --git a/api/tests/unit_tests/core/workflow/nodes/test_document_extractor_node.py b/api/tests/unit_tests/core/workflow/nodes/test_document_extractor_node.py
index 60fa41798f3252..3f169e425730e3 100644
--- a/api/tests/unit_tests/core/workflow/nodes/test_document_extractor_node.py
+++ b/api/tests/unit_tests/core/workflow/nodes/test_document_extractor_node.py
@@ -6,6 +6,7 @@
 
 from core.file import File, FileTransferMethod
 from core.variables import ArrayFileSegment
+from core.variables.variables import StringVariable
 from core.workflow.entities.node_entities import NodeRunResult
 from core.workflow.nodes.document_extractor import DocumentExtractorNode, DocumentExtractorNodeData
 from core.workflow.nodes.document_extractor.document_extractor_node import (
@@ -51,7 +52,9 @@ def test_run_variable_not_found(document_extractor_node, mock_graph_runtime_stat
 
 def test_run_invalid_variable_type(document_extractor_node, mock_graph_runtime_state):
     document_extractor_node.graph_runtime_state = mock_graph_runtime_state
-    mock_graph_runtime_state.variable_pool.get.return_value = "Not an ArrayFileSegment"
+    mock_graph_runtime_state.variable_pool.get.return_value = StringVariable(
+        value="Not an ArrayFileSegment", name="test"
+    )
 
     result = document_extractor_node._run()