diff --git a/flag.views.inc b/flag.views.inc
index bb35a1f..d224021 100644
--- a/flag.views.inc
+++ b/flag.views.inc
@@ -54,12 +54,11 @@ function flag_views_data() {
  */
 function flag_views_data_alter(array &$data) {
   $flags = \Drupal::service('flag')->getFlags();
+  $entity_manager = \Drupal::entityManager();
 
   foreach ($flags as $flag_id => $flag) {
     $entity_type_id = $flag->getFlaggableEntityTypeId();
 
-    $entity_manager = \Drupal::entityManager();
-
     $entity_type = $entity_manager->getDefinition($entity_type_id);
     if ($entity_type->hasHandlerClass('views_data')) {
       $base_table = $entity_manager->getHandler($entity_type_id, 'views_data')->getViewsTableForEntityType($entity_type);