change the restart function