diff --git a/config/partner_gms.mk b/config/partner_gms.mk
index 8c705b45..ee04bd01 100644
--- a/config/partner_gms.mk
+++ b/config/partner_gms.mk
@@ -1,5 +1,5 @@
 ifeq ($(WITH_GMS),true)
-ifeq ($(PRODUCT_IS_ATV),true)
+ifeq ($(WITH_GMS_TV),true)
 $(call inherit-product-if-exists, vendor/partner_gms-tv/products/gms.mk)
 else ifeq ($(WITH_GMS_FI),true)
 $(call inherit-product-if-exists, vendor/partner_gms/products/fi.mk)