diff --git a/pisi/api.py b/pisi/api.py
index 172ad25b..359f3946 100644
--- a/pisi/api.py
+++ b/pisi/api.py
@@ -134,12 +134,8 @@ def finalize():
             ctx.loghandler.flush()
             ctx.log.removeHandler(ctx.loghandler)
 
-        if ctx.repodb:
-            ctx.repodb.finalize()
-            ctx.repodb = None
-        if ctx.installdb:
-            ctx.installdb.finalize()
-            ctx.installdb = None    
+        pisi.db.repodb.finalize()
+        pisi.db.installdb.finalize()
         if ctx.filesdb != None:
             ctx.filesdb.close()
             ctx.filesdb = None