diff --git a/lib/semantic_rails_view_helpers/view_helpers.rb b/lib/semantic_rails_view_helpers/view_helpers.rb
index 327acdd..7af81af 100644
--- a/lib/semantic_rails_view_helpers/view_helpers.rb
+++ b/lib/semantic_rails_view_helpers/view_helpers.rb
@@ -65,8 +65,8 @@ module SemanticRailsViewHelpers
     def semantic_model_data(object)
       SemanticRailsViewHelpers.with_semantic_data do
         type = begin
-                 if object.respond_to?(:model)
-                   object.model.class
+                 if object.respond_to?(:source) and object.respond_to?(:to_source)
+                   object.source.class
                  else
                    object.class
                  end