diff --git a/files_undo_remove.module b/files_undo_remove.module
index 02d8034..a56b495 100644
--- a/files_undo_remove.module
+++ b/files_undo_remove.module
@@ -77,8 +77,8 @@ function files_undo_remove_entity_presave($entity, $entity_type) {
   $instances = field_info_instances($entity_type, $bundle);
   foreach ($instances as $field_name => $field_instance) {
 
-    // Check if the modules are file or image.
-    if (in_array($field_instance['widget']['module'], array('image', 'file'))) {
+    // Check if the modules are file, image or linkimage.
+    if (in_array($field_instance['widget']['module'], array('image', 'file', 'linkimagefield'))) {
       foreach ($entity->{$field_name} as $language => $items) {
         foreach ($items as $delta => $item) {
           // Check if there's a remove_state key on the file values.