models fix
Merge pull request #7 from timjoww/fix

models fix