1 files changed, 6 insertions, 1 deletions
diff --git a/rpkid/rpki/rpkid.py b/rpkid/rpki/rpkid.py
index 8c275688..b923b1a3 100644
--- a/rpkid/rpki/rpkid.py
+++ b/rpkid/rpki/rpkid.py
@@ -102,7 +102,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()