fix failing 2D test