Make sample be proper for 2D