fix the directive to use correct model