Updates to restart action hooks