diff --git a/webform_timeout.module b/webform_timeout.module
index 21ce0fe..ff242d3 100644
--- a/webform_timeout.module
+++ b/webform_timeout.module
@@ -50,7 +50,7 @@ function webform_timeout_theme() {
  *
  * @ingroup form_api
  */
-function webform_timeout_webform_submission_form_alter(array &$form, \Drupal\Core\Form\FormStateInterface $form_state, $form_id) {
+function webform_timeout_webform_submission_form_alter(array &$form, FormStateInterface $form_state, $form_id) {
   if (isset($form['elements']['webform_timeout'])) {
     $access = $form['actions']['submit']['#access'] ?? FALSE;
 
@@ -107,7 +107,7 @@ function webform_timeout_delete_timeout_submitted($form, FormStateInterface &$fo
  * @see \Drupal\webform\WebformSubmissionForm::prepareElements()
  * @see hook_webform_element_ELEMENT_TYPE_alter()
  */
-function webform_timeout_webform_element_alter(array &$element, \Drupal\Core\Form\FormStateInterface $form_state, array $context) {
+function webform_timeout_webform_element_alter(array &$element, FormStateInterface $form_state, array $context) {
   if ($element['#type'] == 'webform_timeout') {
     $form_object = $form_state->getFormObject();
     $webform_submission = $form_object->getEntity();