diff --git a/src/frontend/src/components/core/parameterRenderComponent/components/inputFileComponent/index.tsx b/src/frontend/src/components/core/parameterRenderComponent/components/inputFileComponent/index.tsx
index 689e43641..c6dbb1012 100644
--- a/src/frontend/src/components/core/parameterRenderComponent/components/inputFileComponent/index.tsx
+++ b/src/frontend/src/components/core/parameterRenderComponent/components/inputFileComponent/index.tsx
@@ -44,12 +44,12 @@ export default function InputFileComponent({
 
   function checkFileType(fileName: string): boolean {
     if (fileTypes === undefined) return true;
-    for (let index = 0; index < fileTypes.length; index++) {
-      if (fileName.endsWith(fileTypes[index])) {
-        return true;
-      }
-    }
-    return false;
+
+    // Extract the file extension
+    const fileExtension = fileName.split(".").pop();
+
+    // Check if the extracted extension is in the list of accepted file types
+    return fileTypes.includes(fileExtension || "");
   }
 
   const { mutateAsync, isPending } = usePostUploadFile();