fixes restart button