diff --git a/champss/folding/folding/fold.py b/champss/folding/folding/fold.py
index 5ca1e13b..dde54b97 100644
--- a/champss/folding/folding/fold.py
+++ b/champss/folding/folding/fold.py
@@ -356,7 +356,7 @@ def plot(self, sigma=None, foldpath_plots=None):
             "ap": self.ap,
         }
 
-        SNprof, SN_arr, plot_fname = plot_candidate_archive(
+        SNprof, SN_arr, plot_fname, npz_file = plot_candidate_archive(
             self.archive_fname,
             self.coord_path,
             cand_info=cand_info,
@@ -365,7 +365,7 @@ def plot(self, sigma=None, foldpath_plots=None):
 
         log.info(f"SN of folded profile: {SN_arr}")
 
-        return SNprof, SN_arr, plot_fname
+        return SNprof, SN_arr, plot_fname, npz_file
 
     def cleanup(self):
         """
diff --git a/champss/folding/folding/fold_candidate.py b/champss/folding/folding/fold_candidate.py
index 373477a2..1269f70f 100644
--- a/champss/folding/folding/fold_candidate.py
+++ b/champss/folding/folding/fold_candidate.py
@@ -341,7 +341,7 @@ def main(
             plot_aliases(alias_results, output_path=alias_plot_path)
 
     # Plot
-    SNprof, SN_arr, plot_fname = folder.plot(
+    SNprof, SN_arr, plot_fname, npz_file = folder.plot(
         sigma=sigma, foldpath_plots=foldpath + "/plots/folded_candidate_plots/"
     )
 
@@ -354,6 +354,7 @@ def main(
         "archive_fname": folder.archive_fname,
         "SN": float(SN_arr),
         "path_to_plot": plot_fname,
+        "npz_file": npz_file,
     }
 
     # Update database