fixing model update issue