Restart button fixed