[StateRestoration] formatted code