Fix bug in wish model