fix to incorrect function call for current model