--- a/extensions.st	Sat Nov 08 11:19:40 2014 +0100
+++ b/extensions.st	Sat Nov 08 13:31:49 2014 +0100
@@ -359,22 +359,10 @@
 class: aClass
     "Helper for enabling usage of either real class or RBClass"
 
-    | modelClass |
-
-    ((aClass respondsTo: #isModelClass) and: [ aClass isModelClass ]) ifTrue: [  
-        modelClass := aClass
-    ] ifFalse: [ 
-        aClass isMeta ifTrue: [ 
-            modelClass := self model metaclassNamed: aClass theNonMetaclass name.
-        ] ifFalse: [ 
-            modelClass := self model classNamed: aClass name.
-        ]
-    ].
-
-    self modelClass: modelClass
+    self modelClass: (self model classFor: aClass)
 
     "Created: / 05-10-2014 / 21:04:44 / Jakub Nesveda <nesvejak@fit.cvut.cz>"
-    "Modified: / 06-10-2014 / 07:43:50 / Jakub Nesveda <nesvejak@fit.cvut.cz>"
+    "Modified (comment): / 08-11-2014 / 13:26:35 / Jakub Nesveda <nesvejak@fit.cvut.cz>"
 ! !
 
 !RBMethod methodsFor:'compiling'!