Making sure 2D version works