Fix error with 2D/3D DLB