diff --git a/caracal/workers/selfcal_worker.py b/caracal/workers/selfcal_worker.py
index f7613599..4c492673 100644
--- a/caracal/workers/selfcal_worker.py
+++ b/caracal/workers/selfcal_worker.py
@@ -1198,12 +1198,15 @@ def sofia_mask(trg, num, img_dir, field):
 
         caracal.log.info('++++++++++++++LOADING MASKS+++++++++++++++++')
 
-
+        caracal.log.info(outmaskName)
         datMask = fits.getdata('{}/{}'.format(pipeline.masking,outmaskName))
         # datMask = np.around(datMask.astype(np.float32)).astype(np.int16)
 
         datHead = fits.getheader('{}/{}'.format(pipeline.masking,outmaskName))
         datForn = fits.getdata('{}/{}'.format(pipeline.masking,postGridMaskSof))
+        caracal.log.info(postGridMaskSof)
+        datForn = np.expand_dims(datForn,axis=0)
+        datForn = np.expand_dims(datForn,axis=0)
 
         idxNan = np.isnan(datForn)
         datForn[idxNan] = 0