diff --git a/src/common.js b/src/common.js
index 0210f69489..a16f84fd1b 100644
--- a/src/common.js
+++ b/src/common.js
@@ -450,6 +450,7 @@
             var idx = this._models.indexOf(model);
             if (idx < 0) {
                 this.load_models(false);
+                idx = this._models.indexOf(model);
             }
             var to_load = this._models.slice(
                 Math.max(0, idx - Math.floor(this.batchnum / 2)),