Fixed the mask part