diff --git a/colorbleed/plugins/maya/publish/collect_yeti_rig.py b/colorbleed/plugins/maya/publish/collect_yeti_rig.py
index 3198120374..feb7a19199 100644
--- a/colorbleed/plugins/maya/publish/collect_yeti_rig.py
+++ b/colorbleed/plugins/maya/publish/collect_yeti_rig.py
@@ -26,7 +26,7 @@ class CollectYetiRig(pyblish.api.InstancePlugin):
 
     def process(self, instance):
 
-        assert "input_SET" in cmds.sets(instance.name, query=True), (
+        assert "input_SET" in instance.data["setMembers"], (
             "Yeti Rig must have an input_SET")
 
         # Get the input meshes information