diff --git a/query_performance_gauge.py b/query_performance_gauge.py
index 288ed43..702e8d0 100644
--- a/query_performance_gauge.py
+++ b/query_performance_gauge.py
@@ -5,7 +5,7 @@ from typing import Union
 import mysql.connector.connection
 import trino.dbapi
 
-from connections import get_connection
+from connections import get_connection, clean_up_connection
 
 
 def run_measuring_session(config: dict) -> None:
@@ -31,6 +31,8 @@ def run_measuring_session(config: dict) -> None:
 
     print("Finished the measuring session.")
 
+    clean_up_connection(config["connection_details"])
+
 
 class TestableQuery:
     """