Reimplemented update for the new representation