diff --git a/mail_move_message_models.py b/mail_move_message_models.py
index bfd1e41..94513be 100644
--- a/mail_move_message_models.py
+++ b/mail_move_message_models.py
@@ -33,6 +33,7 @@ class wizard(models.TransientModel):
 
         if 'message_id' in res:
             message = self.env['mail.message'].browse(res['message_id'])
+            res['model'] = message.model
             email_from = message.email_from
             parts = email_split(email_from.replace(' ',','))
             if parts: