Polish restart implementation
Fix a small compilation issue