Fix for restart button