Fix update return