diff --git a/mail_base/static/src/js/base.js b/mail_base/static/src/js/base.js
index d392668..5a43826 100644
--- a/mail_base/static/src/js/base.js
+++ b/mail_base/static/src/js/base.js
@@ -269,8 +269,6 @@ var MailTools = core.Class.extend({
         var msg = {
             id: data.id,
             author_id: data.author_id,
-            needaction_partner_ids: data.needaction_partner_ids,
-            partner_ids: data.partner_ids,
             body_short: data.body_short || "",
             body: data.body || "",
             date: moment(time.str_to_datetime(data.date)),
diff --git a/mail_to/static/src/js/mail_to.js b/mail_to/static/src/js/mail_to.js
index 63b9219..48f3c6e 100644
--- a/mail_to/static/src/js/mail_to.js
+++ b/mail_to/static/src/js/mail_to.js
@@ -1,17 +1,14 @@
 odoo.define('mail_to.MailTo', function (require) {
     "use strict";
 
-    var Thread = require('mail.ChatThread');
-    var Model = require('web.Model');
+    var base_obj = require('mail_base.base');
 
-    Thread.include({
-        render: function (messages, options) {
-            // for(var i = 0; i < messages.length; i++){
-            //     var msg = messages[i];
-            //     msg.needaction_partner_ids = [3];
-            // }
-            // console.log('messages:', messages);
-            this._super(messages, options);
+    base_obj.MailTools.include({
+        make_message: function(data){
+            var msg = this._super(data);
+            msg.partner_ids = data.partner_ids;
+            // msg.needaction_partner_ids = data.needaction_partner_ids;
+            return msg;
         }
     });
 });