Merge pull request #77 from af83/4401-compliance-control-ModelAttribute--address-Robert,s-comments

4401 compliance control model attribute  address robert,s comments

1 files changed, 3 insertions, 2 deletions

diff --git a/lib/model_attribute.rb b/lib/model_attribute.rb
index 60580e306..4d246853a 100644
--- a/lib/model_attribute.rb
+++ b/lib/model_attribute.rb
@@ -12,9 +12,9 @@ class ModelAttribute
   def self.classes
     all
       .map(&:klass)
+      .uniq
       .map(&:to_s)
       .map(&:camelize)
-      .uniq
   end
 
   def self.group_by_class
@@ -93,7 +93,8 @@ class ModelAttribute
   end
 
   def ==(other)
-    klass == other.klass &&
+    self.class === other &&
+      klass == other.klass &&
       name == other.name &&
       data_type == other.data_type
   end