diff --git a/addon/ext.js b/addon/ext.js
index c424399c..c9cba447 100644
--- a/addon/ext.js
+++ b/addon/ext.js
@@ -46,7 +46,7 @@ Store.reopen({
   */
   createFragment: function(modelName, props) {
     Ember.assert("The '" + modelName + "' model must be a subclass of MF.Fragment", this.isFragment(modelName));
-    
+
     var type = this.modelFor(modelName);
     var internalModel = new InternalModel(type, null, this, getOwner(this).container);
 
@@ -296,7 +296,13 @@ function getFragmentTransform(owner, store, attributeType) {
   var polymorphicTypeProp = match[3];
 
   if (!owner.hasRegistration(containerKey)) {
-    var transformClass = owner._lookupFactory('transform:' + transformName);
+    var transformClass;
+    if (owner.factoryFor) {
+      transformClass = owner.factoryFor('transform:' + transformName);
+      transformClass = transformClass && transformClass.class;
+    } else {
+      transformClass = owner._lookupFactory('transform:' + transformName);
+    }
 
     owner.register(containerKey, transformClass.extend({
       store: store,