Fixes to flex_grid.