1 files changed, 6 insertions, 1 deletions
diff --git a/rpkid/rpki/pubd.py b/rpkid/rpki/pubd.py
index 0bf811db..7a00c172 100644
--- a/rpkid/rpki/pubd.py
+++ b/rpkid/rpki/pubd.py
@@ -98,7 +98,12 @@ class main(object):
 
     if self.profile:
       import cProfile
-      cProfile.run("self.main()", self.profile)
+      prof = cProfile.Profile()
+      try:
+        prof.runcall(self.main)
+      finally:
+        prof.dump_stats(self.profile)
+        rpki.log.info("Dumped profile data to %s" % self.profile)
     else:
       self.main()