diff --git a/drivers/falkordb.py b/drivers/falkordb.py
index 697a9ad..87e4dac 100644
--- a/drivers/falkordb.py
+++ b/drivers/falkordb.py
@@ -1,3 +1,4 @@
+import logging
 import os
 from typing import Any
 
@@ -14,9 +15,9 @@ def __init__(self):
 
         # Check if connection is successful
         try:
-            print("FalkorDB database connected successfully!")
+            logging.info("FalkorDB database connected successfully!")
         except ValueError as ve:
-            print("FalkorDB database: {}".format(ve))
+            logging.error("FalkorDB database: {}".format(ve))
             raise
 
     def get_graph_data(self) -> tuple[list[dict[str, Any]], list[dict[str, Any]]]: