Fix widget error for 2D data