diff --git a/api/tests/unit_tests/core/workflow/nodes/llm/test_node.py b/api/tests/unit_tests/core/workflow/nodes/llm/test_node.py
index 519a6cf44e..a77d89ba1e 100644
--- a/api/tests/unit_tests/core/workflow/nodes/llm/test_node.py
+++ b/api/tests/unit_tests/core/workflow/nodes/llm/test_node.py
@@ -8,6 +8,7 @@ from unittest import mock
 
 import pytest
 
+import core.workflow.nodes.llm.node
 from core.app.entities.app_invoke_entities import InvokeFrom, ModelConfigWithCredentialsEntity
 from core.entities.provider_configuration import ProviderConfiguration, ProviderModelBundle
 from core.entities.provider_entities import CustomConfiguration, SystemConfiguration
@@ -37,7 +38,7 @@ from core.workflow.nodes.llm.entities import (
     VisionConfigOptions,
 )
 from core.workflow.nodes.llm.file_saver import LLMFileSaver
-from core.workflow.nodes.llm.node import LLM_NODE_THINKING_TAGS_ENABLED, LLMNode
+from core.workflow.nodes.llm.node import LLMNode
 from models.enums import UserFrom
 from models.provider import ProviderType
 from models.workflow import WorkflowType
@@ -751,17 +752,17 @@ with multiple lines
     def test_environment_variable_enabled(self):
         """Test that environment variable is properly read when enabled."""
         importlib.reload(core.workflow.nodes.llm.node)
-        assert LLM_NODE_THINKING_TAGS_ENABLED is True
+        assert core.workflow.nodes.llm.node.LLM_NODE_THINKING_TAGS_ENABLED is True
 
     @mock.patch.dict("os.environ", {"LLM_NODE_THINKING_TAGS_ENABLED": "false"})
     def test_environment_variable_disabled(self):
         """Test that environment variable is properly read when disabled."""
         importlib.reload(core.workflow.nodes.llm.node)
-        assert LLM_NODE_THINKING_TAGS_ENABLED is False
+        assert core.workflow.nodes.llm.node.LLM_NODE_THINKING_TAGS_ENABLED is False
 
     def test_environment_variable_default(self):
         """Test that environment variable defaults to True."""
         with mock.patch.dict("os.environ"):
             os.environ.pop("LLM_NODE_THINKING_TAGS_ENABLED", None)
             importlib.reload(core.workflow.nodes.llm.node)
-            assert LLM_NODE_THINKING_TAGS_ENABLED is True
+            assert core.workflow.nodes.llm.node.LLM_NODE_THINKING_TAGS_ENABLED is True