diff --git a/tests/data/component_with_templatefield.py b/tests/data/component_with_templatefield.py
index 725bec9b2..6b2ce011b 100644
--- a/tests/data/component_with_templatefield.py
+++ b/tests/data/component_with_templatefield.py
@@ -1,7 +1,7 @@
 import random
 
 from langflow import CustomComponent
-from langflow.template.field.base import TemplateField
+from langflow.field_typing import TemplateField
 
 
 class TestComponent(CustomComponent):