fixing restart button