Add max2d fix