Fix restart magic test
Add restart magic test