It would be more consistent if you used `np.ma.array` here like you do later.