diff --git a/file_entity.install b/file_entity.install
index 334cb5c..e8b503a 100644
--- a/file_entity.install
+++ b/file_entity.install
@@ -52,6 +52,8 @@ function file_entity_schema() {
 function file_entity_install() {
   $type_storage_definition = \Drupal::entityManager()->getFieldStorageDefinitions('file')['type'];
   \Drupal::entityManager()->getStorage('file')->onFieldStorageDefinitionCreate($type_storage_definition);
+  \Drupal::entityDefinitionUpdateManager()
+    ->installFieldStorageDefinition('type', 'file', 'file_entity', $type_storage_definition);
 
   // Set permissions.
   $roles = user_roles();
@@ -87,7 +89,11 @@ function file_entity_install() {
 function file_entity_uninstall() {
   // Remove the added column to the core {file_managed} table.
   db_drop_field('file_managed', 'type');
+  $update_manager = \Drupal::entityDefinitionUpdateManager();
+  $type_storage_definition = $update_manager->getFieldStorageDefinition('type', 'file');
+  $update_manager->uninstallFieldStorageDefinition($type_storage_definition);
 }
+
 /**
  * Update existing bogus entries for File types Document.
  */