The blow up issue is solved by correcting the Ez update function