From 26a353f118ed3cce01e845975cb43a195fd18f21 Mon Sep 17 00:00:00 2001
From: Thibault Barnouin <thibault.barnouin@astro.unistra.fr>
Date: Mon, 14 Apr 2025 13:25:21 +0200
Subject: [PATCH] change header name for Stat error

---
 package/lib/fits.py  | 4 ++--
 package/lib/plots.py | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/package/lib/fits.py b/package/lib/fits.py
index 62425f4..43d62d9 100755
--- a/package/lib/fits.py
+++ b/package/lib/fits.py
@@ -213,10 +213,10 @@ def save_Stokes(
         [P, "Pol_deg"],
         [debiased_P, "Pol_deg_debiased"],
         [s_P, "Pol_deg_err"],
-        [s_P_P, "Pol_deg_err_Poisson_noise"],
+        [s_P_P, "Pol_deg_stat_err"],
         [PA, "Pol_ang"],
         [s_PA, "Pol_ang_err"],
-        [s_PA_P, "Pol_ang_err_Poisson_noise"],
+        [s_PA_P, "Pol_ang_stat_err"],
         [data_mask, "Data_mask"],
     ]:
         hdu_header = header.copy()
diff --git a/package/lib/plots.py b/package/lib/plots.py
index e428d28..2dfdb28 100755
--- a/package/lib/plots.py
+++ b/package/lib/plots.py
@@ -2978,7 +2978,7 @@ class pol_map(object):
         return self.Stokes["DATA_MASK"].data
 
     def set_data_mask(self, mask):
-        self.Stokes[np.argmax([self.Stokes[i].header["datatype"] == "Data_mask" for i in range(len(self.Stokes))])].data = mask.astype(float)
+        self.Stokes["DATA_MASK"].data = mask.astype(float)
 
     @property
     def cut(self):