This doesn't look elegant to me, but at least it works for now.
It might be better to just use YaspGrid directly in the future.