Fix re-shaping drag block issue