Fixed problem with grid flow