Update to keep callback