diff --git a/node_type_class.module b/node_type_class.module
index a739e82..a5f9024 100644
--- a/node_type_class.module
+++ b/node_type_class.module
@@ -1,5 +1,8 @@
 <?php
 
+use Drupal\node\NodeTypeInterface;
+use Drupal\Core\Form\FormStateInterface;
+
 /**
  * @file
  * This is the file description for Node type class module.
@@ -11,7 +14,7 @@
 /**
  * Implements hook_form_FORM_ID_alter().
  */
-function node_type_class_form_node_type_edit_form_alter(&$form, \Drupal\Core\Form\FormStateInterface &$form_state) {
+function node_type_class_form_node_type_edit_form_alter(&$form, FormStateInterface &$form_state) {
   /** @var \Drupal\node\NodeTypeInterface $type */
   $type = $form_state->getFormObject()->getEntity();
   $classes = $type->getThirdPartySetting('node_type_class', 'classes', '');
@@ -37,7 +40,7 @@ function node_type_class_form_node_type_edit_form_alter(&$form, \Drupal\Core\For
 /**
  * Entity builder for the node type form with Node type class.
  */
-function node_type_class_form_node_type_form_builder($entity_type, NodeTypeInterface $type, &$form, \Drupal\Core\Form\FormStateInterface $form_state) {
+function node_type_class_form_node_type_form_builder($entity_type, NodeTypeInterface $type, &$form, FormStateInterface $form_state) {
   $type->setThirdPartySetting('node_type_class', 'classes', $form_state->getValue('node_type_class'));
 }