diff --git a/Libs/DF/picture.lua b/Libs/DF/picture.lua
index 8af78f41..4bd6d28c 100644
--- a/Libs/DF/picture.lua
+++ b/Libs/DF/picture.lua
@@ -300,7 +300,7 @@ detailsFramework:Mixin(ImageMetaFunctions, detailsFramework.ScriptHookMixin)
 		if (texture) then
 			if (type(texture) == "table") then
 				if (texture.gradient) then
-					if (detailsFramework.IsDragonflight() or ImageObject.image.SetGradient) then
+					if (detailsFramework.IsDragonflight() or detailsFramework.IsWotLKWowWithRetailAPI()) then
 						ImageObject.image:SetColorTexture(1, 1, 1, 1)
 						local fromColor = detailsFramework:FormatColor("tablemembers", texture.fromColor)
 						local toColor = detailsFramework:FormatColor("tablemembers", texture.toColor)
@@ -343,4 +343,4 @@ detailsFramework:Mixin(ImageMetaFunctions, detailsFramework.ScriptHookMixin)
 		setmetatable(ImageObject, ImageMetaFunctions)
 
 		return ImageObject
-	end
\ No newline at end of file
+	end