diff --git a/fieldblock.module b/fieldblock.module
index dfa6d57c8f9a10d6b17509cc868fd373445a6ba4..8dc95091f3254c6b06382159f719d6ac4071dc26 100644
--- a/fieldblock.module
+++ b/fieldblock.module
@@ -11,7 +11,7 @@ use Drupal\Core\Entity\Display\EntityViewDisplayInterface;
 
 /**
  * Implements hook_form_alter().
- * 
+ *
  * Adds a column to the "display fields" table-form, with a checkbox for each
  * field.
  */
@@ -81,11 +81,11 @@ function fieldblock_field_display_submit($form, FormStateInterface $form_state)
  * Takes fields out of the current entity and caches them in a post render cache
  * context. The #post_render_cache callback makes this data available to the
  * fieldblock when it is built, We also hide the field from the render array.
- * 
+ *
  * @see \Drupal\fieldblock\Plugin\Block\FieldBlock::fieldBlockPostRenderCache
  * @see https://www.drupal.org/node/2151609
  */
-function fieldblock_entity_view_alter(array &$build, Drupal\Core\Entity\EntityInterface $entity, EntityViewDisplayInterface $display) {
+function fieldblock_entity_view_alter(array &$build, $entity, EntityViewDisplayInterface $display) {
   $fieldblock_settings = $display->getThirdPartySettings('fieldblock');
   $display_id = $display->get('id');