Forgot to update the models previously
Merge pull request #22 from cheerfulstoic/patch-3

Forgot to update the models previously