Adding update to get uniform behavoir