Update for FormStateInterface