Small Fix in models