Bugfix for masked z-values with imshow