diff --git a/core/lib/Drupal/Core/Entity/EntityManager.php b/core/lib/Drupal/Core/Entity/EntityManager.php
index 7a867b8..01cf768 100644
--- a/core/lib/Drupal/Core/Entity/EntityManager.php
+++ b/core/lib/Drupal/Core/Entity/EntityManager.php
@@ -240,7 +240,8 @@ protected function processDefinition(&$definition, $plugin_id) {
     foreach ($view_modes as $config_name) {
       $view_mode = config($config_name)->get();
       if ($view_mode['type'] == $plugin_id) {
-        $definition['view_modes'][$view_mode['machine_name']] = array(
+        $entity_type = strtok($view_mode['machine_name'], '.');
+        $definition['view_modes'][strtok('.')] = array(
           'label' => check_plain($view_mode['name']),
           'custom_settings' => $view_mode['custom'],
         );