Forgot to update the models previously

Sorry for the second pull request, I was doing this through the github UI.  Should probably do it on the command line next time ;)