diff --git a/widgets/ItemButton.lua b/widgets/ItemButton.lua
index e7f93cde..2b0c3c03 100644
--- a/widgets/ItemButton.lua
+++ b/widgets/ItemButton.lua
@@ -75,7 +75,7 @@ else
 	buttonClass, buttonProto = addon:NewClass("ItemButton", "Button", "ContainerFrameItemButtonTemplate", "ABEvent-1.0")
 end
 
-local childrenNames = { "Cooldown", "IconTexture", "IconQuestTexture", "Count", "Stock", "NormalTexture", "NewItemTexture" }
+local childrenNames = { "Cooldown", "IconTexture", "IconQuestTexture", "Count", "Stock", "NormalTexture", "NewItemTexture", "IconOverlay2" }
 
 function buttonProto:OnCreate()
 	local name = self:GetName()
@@ -327,6 +327,7 @@ function buttonProto:Update()
 	self:UpdateNew()
 	if addon.isRetail then
 		self:UpdateUpgradeIcon()
+		self:UpdateOverlay()
 	end
 	if self.UpdateSearch then
 		self:UpdateSearch()
@@ -439,6 +440,11 @@ function buttonProto:UpdateBorder(isolatedEvent)
 	end
 end
 
+function buttonProto:UpdateOverlay()
+	local _, _, _, quality = GetContainerItemInfo(self.bag, self.slot)
+	SetItemButtonOverlay(self, self.itemId or self.itemLink or 0, quality)
+end
+
 --------------------------------------------------------------------------------
 -- Item stack button
 --------------------------------------------------------------------------------