diff --git a/src/backend/langflow/interface/custom/component.py b/src/backend/langflow/interface/custom/component.py
index 0f503accb..98ccf8db3 100644
--- a/src/backend/langflow/interface/custom/component.py
+++ b/src/backend/langflow/interface/custom/component.py
@@ -4,7 +4,6 @@ from typing import Any, ClassVar, Optional
 
 from cachetools import TTLCache, cachedmethod
 from fastapi import HTTPException
-
 from langflow.interface.custom.code_parser import CodeParser
 from langflow.utils import validate
 
@@ -30,7 +29,7 @@ class Component:
         for key, value in data.items():
             setattr(self, key, value)
 
-    @cachedmethod(operator.attrgetter("cache"))
+    @cachedmethod(cache=operator.attrgetter("cache"))
     def get_code_tree(self, code: str):
         parser = CodeParser(code)
         return parser.parse_code()