Replace FormState with FormStateInterface