Cost model fix