Merge aa8571d into e28afa4

astropy · Jun 13, 2016 · bf804bd · bf804bd
2 parents e28afa4 + aa8571d
commit bf804bd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/astroscrappy/astroscrappy.pyx b/astroscrappy/astroscrappy.pyx
@@ -338,7 +338,8 @@ def detect_cosmics(indat, inmask=None, float sigclip=4.5, float sigfrac=0.3,
         # otherwise clean the image and iterate
         if cleantype == 'median':
             # Unmasked median filter
-            cleanarr[crmask] = m5[crmask]
+            cindices = crmask.nonzero()
+            cleanarr[cindices] = m5[cindices]
             del m5
         # Masked mean filter
         elif cleantype == 'meanmask':