fix for restart