From 3b2861664042b85678357fb50b3fad34a7bc209a Mon Sep 17 00:00:00 2001
From: Joel T <joel.thiessen@automattic.com>
Date: Mon, 25 Apr 2022 14:30:29 -0700
Subject: [PATCH] Altering badge logic to function based on experiments instead
 of feature flag

---
 .../src/Admin/Features/OnboardingTasks/TaskLists.php       | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/plugins/woocommerce/src/Admin/Features/OnboardingTasks/TaskLists.php b/plugins/woocommerce/src/Admin/Features/OnboardingTasks/TaskLists.php
index f6434a87313..6ee958d431e 100644
--- a/plugins/woocommerce/src/Admin/Features/OnboardingTasks/TaskLists.php
+++ b/plugins/woocommerce/src/Admin/Features/OnboardingTasks/TaskLists.php
@@ -468,11 +468,14 @@ class TaskLists {
 	 * @return number
 	 */
 	public static function setup_tasks_remaining () {
-		if( ! Features::is_enabled( 'tasklist-setup-experiment-1' )) {
+
+		$active_list = self::is_experiment_treatment( 'woocommerce_tasklist_setup_experiment_1' ) ? 'setup_experiment_1' : self::is_experiment_treatment( 'woocommerce_tasklist_setup_experiment_2' ) ? 'setup_experiment_1' : null;
+
+		if( is_null( $active_list ) ) {
 			return;
 		}
 
-		$setup_list = self::get_list( 'setup_experiment_1' );
+		$setup_list = self::get_list( $active_list );
 
 		if( $setup_list->is_hidden() || $setup_list->is_complete() ) {
 			return;