diff --git a/warden_server/test_warden_server.py b/warden_server/test_warden_server.py
index bee3433cf9627751788fb6c779914846b5133398..e76e35645e926c7536c472f87bcd7ea1948dccbe 100755
--- a/warden_server/test_warden_server.py
+++ b/warden_server/test_warden_server.py
@@ -412,7 +412,7 @@ def init_user():
     conn = None
     try:
         conn = my.connect(user='root', passwd=getpass.getpass('Enter MySQL Root password:'))
-        with conn as cur:   # Not a canonical connector implementation, for sure
+        with conn.cursor() as cur:
             cur.execute("GRANT SELECT, INSERT, UPDATE, CREATE, DELETE, DROP ON *.* TO %s@'localhost' IDENTIFIED BY %s", (USER, PASSWORD))
         conn.commit()
         print("DB User set up successfuly")