Use the new $form_state['controller'] to fetch the form entity.