diff --git a/sites/all/modules/media/media.install b/sites/all/modules/media/media.install
index e3a81d9..ae24453 100644
--- a/sites/all/modules/media/media.install
+++ b/sites/all/modules/media/media.install
@@ -636,7 +636,8 @@ function media_update_7017() {
   // configurations. However, if file_entity_update_7001() has already run
   // prior to media_update_7016(), run it again in order to capture those
   // configurations too.
-  if (db_table_exists('file_display')) {    
+  if (db_table_exists('file_display')) {
+    module_load_include('install', 'file_entity', 'file_entity');
     file_entity_update_7001();
   }
 }