diff --git a/drivers/watchdogtest/watchdogtest.c b/drivers/watchdogtest/watchdogtest.c
index 1bc434a13f58..50fc698e335e 100644
--- a/drivers/watchdogtest/watchdogtest.c
+++ b/drivers/watchdogtest/watchdogtest.c
@@ -75,14 +75,16 @@ static int watchdog_test_init(void)
 #if LINUX_VERSION_CODE < KERNEL_VERSION(5,4,0)
 	trigger_wdog_bite = (void *)kallsyms_lookup_name("msm_trigger_wdog_bite");
 #else
-#ifdef CONFIG_ARCH_QCOM
+#if defined(CONFIG_QCOM_WATCHDOG_V2)
 	trigger_wdog_bite = msm_trigger_wdog_bite;
+#elif defined(CONFIG_QCOM_WDT_CORE)
+	trigger_wdog_bite = qcom_wdt_trigger_bite;
 #else
 	trigger_wdog_bite = NULL;
 #endif
 #endif
 	if (!trigger_wdog_bite)
-		printk(KERN_ERR "Failed to get msm_trigger_wdog_bite address\n");
+		printk(KERN_ERR "Failed to get trigger_wdog_bite address\n");
 	return 0;
 }