Restore button update. Fixes gh-11