production: Use update code instead of duplicating it