diff --git a/Setup/UpgradeData.php b/Setup/UpgradeData.php
index 66385b03..59f3be85 100644
--- a/Setup/UpgradeData.php
+++ b/Setup/UpgradeData.php
@@ -98,7 +98,7 @@ private function removeProductAttributesAndSets()
         }
 
         $entityTypeId = $categorySetup->getEntityTypeId(Product::ENTITY);
-        $attributeSetId = $categorySetup->getAttributeSetId($entityTypeId, self::VINDI_PLANOS);
+        $attributeSetId = $categorySetup->getAttributeSet($entityTypeId, self::VINDI_PLANOS, 'attribute_set_id');
         if ($attributeSetId) {
             $attributeGroupId = $categorySetup->getAttributeGroupId($entityTypeId, $attributeSetId, self::VINDI_PLAN_SETTINGS);
             if ($attributeGroupId) {