Update to use restart instead of WRT.restart