fix small bug to get the grid to work round 2