Ah yes, I can see that... I guess @stefanv was looking at the argument name. Maybe change it to `tiling_shape`? @stefanv, what would you suggest?