diff --git a/Android.mk b/Android.mk
index 8391e02..1c2650b 100644
--- a/Android.mk
+++ b/Android.mk
@@ -25,9 +25,9 @@
 LOCAL_SRC_FILES := thermal.c
 LOCAL_SRC_FILES += thermal_common.c
 
-ifeq ($(call is-board-platform-in-list,msm8998), true)
+ifneq (,$(call is-board-platform-in-list2,msm8998))
 LOCAL_SRC_FILES += thermal-8998.c
-else ifeq ($(call is-board-platform-in-list,sdm845), true)
+else ifneq (,$(call is-board-platform-in-list2,sdm845))
 LOCAL_SRC_FILES += thermal-845.c
 else
 LOCAL_SRC_FILES += thermal_target.c