diff --git a/Koha/Schema/Result/EdifactMessage.pm b/Koha/Schema/Result/EdifactMessage.pm
index db24e7468c..9bfd3c51ab 100644
--- a/Koha/Schema/Result/EdifactMessage.pm
+++ b/Koha/Schema/Result/EdifactMessage.pm
@@ -202,4 +202,11 @@ __PACKAGE__->add_columns(
     '+deleted' => { is_boolean => 1 },
 );
 
+sub koha_objects_class {
+    'Koha::Edifact::Files';
+}
+sub koha_object_class {
+    'Koha::Edifact::File';
+}
+
 1;