Update to create_grid.py

Proper fix for the previous error.