Fix and cleanup models.py after removing WeakForeignKey

diff --git a/telemeta/models/core.py b/telemeta/models/core.py
index 8ac633a64a0071ee8bf076cd40324afe53de81dc..607cec0b984ae1ebd478ea4eee89da27384af418 100644 (file)
--- a/telemeta/models/core.py
+++ b/telemeta/models/core.py
@@ -114,7 +114,7 @@ class ModelCore(models.Model, DirtyFieldsMixin):
         fields = self.to_dict()
         for name, value in fields.iteritems():
             element = doc.createElement(self.get_dom_field_name(name))
-            if isinstance(value, EnhancedModel):
+            if isinstance(value, ModelCore):
                 element.setAttribute('key', str(value.pk))
             value = unicode(value)
             element.appendChild(doc.createTextNode(value))