update calendar build_links for table builder

1 files changed, 2 insertions, 2 deletions

diff --git a/app/helpers/table_builder_helper/url.rb b/app/helpers/table_builder_helper/url.rb
index 40e1f839f..0e3dce0aa 100644
--- a/app/helpers/table_builder_helper/url.rb
+++ b/app/helpers/table_builder_helper/url.rb
@@ -3,7 +3,7 @@ module TableBuilderHelper
     def self.polymorphic_url_parts(item, referential, workgroup)
       polymorph_url = []
 
-      unless item.is_a?(Referential) || item.is_a?(ComplianceControlSet)
+      unless item.is_a?(Calendar) || item.is_a?(Referential) || item.is_a?(ComplianceControlSet)
         if referential
           polymorph_url << referential
           polymorph_url << item.line if item.respond_to? :line
@@ -20,7 +20,7 @@ module TableBuilderHelper
           end
         end
       else
-        polymorph_url << item.workgroup if item.is_a?(Calendar)
+        polymorph_url << item.workgroup if item.respond_to? :workgroup
         polymorph_url << item
       end