Fix bug model