diff --git a/api/core/variables/variables.py b/api/core/variables/variables.py
index f9268b52e6c490..973e420961bb46 100644
--- a/api/core/variables/variables.py
+++ b/api/core/variables/variables.py
@@ -90,5 +90,5 @@ class FileVariable(FileSegment, Variable):
     pass
 
 
-class ArrayFileVariable(ArrayFileSegment, Variable):
+class ArrayFileVariable(ArrayFileSegment, ArrayVariable):
     pass
diff --git a/api/tests/unit_tests/core/app/segments/test_variables.py b/api/tests/unit_tests/core/app/segments/test_variables.py
index 0c264c15a03593..426557c7161419 100644
--- a/api/tests/unit_tests/core/app/segments/test_variables.py
+++ b/api/tests/unit_tests/core/app/segments/test_variables.py
@@ -2,6 +2,8 @@
 from pydantic import ValidationError
 
 from core.variables import (
+    ArrayFileVariable,
+    ArrayVariable,
     FloatVariable,
     IntegerVariable,
     ObjectVariable,
@@ -81,3 +83,8 @@ def test_variable_to_object():
     assert var.to_object() == 3.14
     var = SecretVariable(name="secret", value="secret_value")
     assert var.to_object() == "secret_value"
+
+
+def test_array_file_variable_is_array_variable():
+    var = ArrayFileVariable(name="files", value=[])
+    assert isinstance(var, ArrayVariable)