Whoops. I missed out the model callbacks.
Everything seems to be working now at the most basic level.