diff --git a/package/FOC_reduction.py b/package/FOC_reduction.py
index c2da366..414a8c7 100755
--- a/package/FOC_reduction.py
+++ b/package/FOC_reduction.py
@@ -274,7 +274,7 @@ def main(target=None, proposal_id=None, infiles=None, output_dir="./data", crop=
         flux_data=np.array([flux_data, flux_error, flux_mask]),
         flux_head=flux_head,
     )
-    outfiles.append("/".join([data_folder, Stokes_hdul["I_STOKES"].header["FILENAME"] + ".fits"]))
+    outfiles.append("/".join([data_folder, Stokes_hdul[0].header["FILENAME"] + ".fits"]))
 
     # Step 5:
     # crop to desired region of interest (roi)
diff --git a/package/lib/plots.py b/package/lib/plots.py
index b98beac..346df15 100755
--- a/package/lib/plots.py
+++ b/package/lib/plots.py
@@ -2951,7 +2951,19 @@ class pol_map(object):
 
     @property
     def wcs(self):
-        return WCS(self.Stokes[0].header).celestial.deepcopy()
+        return WCS(self.Stokes["I_STOKES"].header).celestial.deepcopy()
+
+    @property
+    def Flux(self):
+        return self.Stokes[0].data[0] * self.Stokes[0].header["PHOTFLAM"]
+
+    @property
+    def Flux_err(self):
+        return self.Stokes[0].data[1] * self.Stokes[0].header["PHOTFLAM"]
+
+    @property
+    def Flux_mask(self):
+        return self.Stokes[0].data[2].astype(bool)
 
     @property
     def Flux(self):