From 6523f9587f29089304985a214e5018769405dd5e Mon Sep 17 00:00:00 2001
From: John Bintz <john@coswellproductions.com>
Date: Sun, 24 Feb 2013 17:51:24 -0500
Subject: [PATCH] better decorator detection

---
 lib/semantic_rails_view_helpers/view_helpers.rb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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