1 files changed, 15 insertions, 19 deletions

diff --git a/node_modules/montage/ui/application.js b/node_modules/montage/ui/application.js
index 43443ecc..025ef950 100755
--- a/node_modules/montage/ui/application.js
+++ b/node_modules/montage/ui/application.js
@@ -137,31 +137,27 @@ var Application = exports.Application = Montage.create(Montage, /** @lends monta
         value: null
     },
 
-    /**
-     Description TODO
-     @function
-     @param {Function} callback A function to invoke after the method has completed.
-     */
-    load: {
-        value: function(callback) {
-            var template = Template.create().initWithDocument(window.document),
-                component,
+    _load: {
+        value: function(applicationRequire, callback) {
+            var template = Template.create().initWithDocument(window.document, applicationRequire),
+                rootComponent,
                 self = this;
 
-            self = Application.isPrototypeOf(self) ? self : Application.create();
-
             // assign to the exports so that it is available in the deserialization of the template
             exports.application = self;
 
-            ///// TODO: Remove this hack when montage exposes this
-            this._template = template;
+            require.async("ui/component").then(function(exports) {
+                rootComponent = exports.__root__;
+                rootComponent.element = document;
+                template.instantiateWithOwnerAndDocument(null, window.document, function() {
+                    self.callDelegateMethod("willFinishLoading", self);
+                    rootComponent.needsDraw = true;
+                    if (callback) {
+                        callback(self);
+                    }
 
-            template.instantiateWithOwnerAndDocument(null, window.document, function() {
-                require("ui/component").__root__.needsDraw = true;
-                if (callback) {
-                    callback(self);
-                }
-            });
+                });
+            }).end();
         }
     },