fix to incorrect function call for current model
Merge pull request #37 from edmz/patch-1

fix to incorrect function call for current model