Bug fix: init the boundary condition for 2D