fix the grid demo