updated build

nielsgl · May 2, 2016 · a72e5d8 · a72e5d8
1 parent 66d1bff
commit a72e5d8
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/lib/index.js b/lib/index.js
@@ -170,7 +170,7 @@ exports.default = function (sequelize, options) {
       this.addHook("afterUpdate", afterHook);
 
       // create association
-      this.hasMany(sequelize.models.Revision, {
+      this.hasMany(sequelize.models[options.revisionModel], {
         foreignKey: "document_id",
         constraints: false,
         scope: {
@@ -286,7 +286,7 @@ exports.default = function (sequelize, options) {
 
           d.save().then(function (d) {
             // Add diff to revision
-            revision.addRevisionChange(d);
+            revision['add' + options.revisionChangeModel](d);
             return null;
           }).catch(function (err) {
             log('RevisionChange save error');
@@ -372,6 +372,7 @@ exports.default = function (sequelize, options) {
       var RevisionChange = sequelize.define(options.revisionChangeModel, attributes, {
         underscored: options.underscored
       });
+
       // Set associations
       Revision.hasMany(RevisionChange, {
         foreignKey: options.defaultAttributes.revisionId,