Potential issue in set_grid

Just not to lose track of this potential issue @nschorgh mentioned here [https://github.com/sampotter/python-flux/pull/21#issuecomment-809829387](https://github.com/sampotter/python-flux/pull/21#issuecomment-809829387) (easy enough to solve, but since we are in the middle of several PR already...)