fix bug 2d version