diff --git a/src/backend/langflow/interface/custom/custom_component/custom_component.py b/src/backend/langflow/interface/custom/custom_component/custom_component.py
index 15e7cd845..f946c7695 100644
--- a/src/backend/langflow/interface/custom/custom_component/custom_component.py
+++ b/src/backend/langflow/interface/custom/custom_component/custom_component.py
@@ -5,7 +5,6 @@ from uuid import UUID
 import yaml
 from cachetools import TTLCache, cachedmethod
 from fastapi import HTTPException
-
 from langflow.interface.custom.code_parser.utils import (
     extract_inner_type_from_generic_alias,
     extract_union_types_from_generic_alias,
@@ -174,7 +173,6 @@ class CustomComponent(Component):
 
         return get_index
 
-    @property
     def get_function(self):
         return validate.create_function(self.code, self.function_entrypoint_name)
 
diff --git a/tests/test_process.py b/tests/test_process.py
index 29ec267a1..c8e4ec9cc 100644
--- a/tests/test_process.py
+++ b/tests/test_process.py
@@ -1,3 +1,4 @@
+import pytest
 from langflow.processing.process import process_tweaks
 from langflow.services.deps import get_session_service
 
@@ -197,39 +198,42 @@ def test_tweak_not_in_template():
     assert result == graph_data
 
 
-def test_load_langchain_object_with_cached_session(client, basic_graph_data):
+@pytest.mark.asyncio
+async def test_load_langchain_object_with_cached_session(client, basic_graph_data):
     # Provide a non-existent session_id
     session_service = get_session_service()
     session_id1 = "non-existent-session-id"
-    graph1, artifacts1 = session_service.load_session(session_id1, basic_graph_data)
+    graph1, artifacts1 = await session_service.load_session(session_id1, basic_graph_data)
     # Use the new session_id to get the langchain_object again
-    graph2, artifacts2 = session_service.load_session(session_id1, basic_graph_data)
+    graph2, artifacts2 = await session_service.load_session(session_id1, basic_graph_data)
 
     assert graph1 == graph2
     assert artifacts1 == artifacts2
 
 
-def test_load_langchain_object_with_no_cached_session(client, basic_graph_data):
+@pytest.mark.asyncio
+async def test_load_langchain_object_with_no_cached_session(client, basic_graph_data):
     # Provide a non-existent session_id
     session_service = get_session_service()
     session_id1 = "non-existent-session-id"
     session_id = session_service.build_key(session_id1, basic_graph_data)
-    graph1, artifacts1 = session_service.load_session(session_id, basic_graph_data)
+    graph1, artifacts1 = await session_service.load_session(session_id, basic_graph_data)
     # Clear the cache
     session_service.clear_session(session_id)
     # Use the new session_id to get the langchain_object again
-    graph2, artifacts2 = session_service.load_session(session_id, basic_graph_data)
+    graph2, artifacts2 = await session_service.load_session(session_id, basic_graph_data)
 
     assert id(graph1) != id(graph2)
     # Since the cache was cleared, objects should be different
 
 
-def test_load_langchain_object_without_session_id(client, basic_graph_data):
+@pytest.mark.asyncio
+async def test_load_langchain_object_without_session_id(client, basic_graph_data):
     # Provide a non-existent session_id
     session_service = get_session_service()
     session_id1 = None
-    graph1, artifacts1 = session_service.load_session(session_id1, basic_graph_data)
+    graph1, artifacts1 = await session_service.load_session(session_id1, basic_graph_data)
     # Use the new session_id to get the langchain_object again
-    graph2, artifacts2 = session_service.load_session(session_id1, basic_graph_data)
+    graph2, artifacts2 = await session_service.load_session(session_id1, basic_graph_data)
 
     assert graph1 == graph2