Fix model refs
Merge pull request #78 from raymondfeng/master

Fix model refs