This seems to be the set of changes (alongside changing some of the `_build` versus `_update`) that cause the error. If I add `self._update()` immediately after this line, the demo works again. I don't know if that's correct though because I don't completely understand the flow here.