diff --git a/tests/data/component_multiple_outputs.py b/tests/data/component_multiple_outputs.py
index 5cdff8773..e0a5f6846 100644
--- a/tests/data/component_multiple_outputs.py
+++ b/tests/data/component_multiple_outputs.py
@@ -1,11 +1,12 @@
 from langflow.custom import Component
-from langflow.template.field.base import Input, Output
+from langflow.inputs.inputs import IntInput, TextInput
+from langflow.template.field.base import Output
 
 
 class MultipleOutputsComponent(Component):
     inputs = [
-        Input(display_name="Input", name="input", field_type=str),
-        Input(display_name="Number", name="number", field_type=int),
+        TextInput(display_name="Input", name="input"),
+        IntInput(display_name="Number", name="number"),
     ]
     outputs = [
         Output(display_name="Certain Output", name="certain_output", method="certain_output"),
diff --git a/tests/data/component_nested_call.py b/tests/data/component_nested_call.py
index 6cf64944a..8dbc33cc4 100644
--- a/tests/data/component_nested_call.py
+++ b/tests/data/component_nested_call.py
@@ -1,13 +1,14 @@
 from random import randint
 
 from langflow.custom import Component
-from langflow.template.field.base import Input, Output
+from langflow.inputs.inputs import IntInput, TextInput
+from langflow.template.field.base import Output
 
 
 class MultipleOutputsComponent(Component):
     inputs = [
-        Input(display_name="Input", name="input", field_type=str),
-        Input(display_name="Number", name="number", field_type=int),
+        TextInput(display_name="Input", name="input"),
+        IntInput(display_name="Number", name="number"),
     ]
     outputs = [
         Output(display_name="Certain Output", name="certain_output", method="certain_output"),