Small bug fix in grid