1 files changed, 8 insertions, 2 deletions
diff --git a/app/models/workbench.rb b/app/models/workbench.rb
index d99197b47..64e9ae92d 100644
--- a/app/models/workbench.rb
+++ b/app/models/workbench.rb
@@ -51,8 +51,14 @@ class Workbench < ApplicationModel
     where(name: DEFAULT_WORKBENCH_NAME).last
   end
 
-  def import_compliance_control_set
-    import_compliance_control_set_id && ComplianceControlSet.find(import_compliance_control_set_id)
+  # XXX
+  # def import_compliance_control_set
+  #   import_compliance_control_set_id && ComplianceControlSet.find(import_compliance_control_set_id)
+  # end
+
+  def compliance_control_set key
+    id = (owner_compliance_control_set_ids || {})[key.to_s]
+    id.present? && ComplianceControlSet.find(id)
   end
 
   private