diff --git a/tests/System/TasksTest.php b/tests/System/TasksTest.php
index 1e15bac..4adcf46 100644
--- a/tests/System/TasksTest.php
+++ b/tests/System/TasksTest.php
@@ -13,6 +13,7 @@
 use Piwik\Piwik;
 use Piwik\Plugins\AnonymousPiwikUsageMeasurement\tests\Fixtures\SendSystemReportTaskFixture;
 use Piwik\Tests\Framework\TestCase\SystemTestCase;
+use Piwik\Version;
 
 /**
  * @group AnonymousPiwikUsageMeasurement
@@ -79,6 +80,12 @@ public function getApiForTesting()
             'CustomVariables',
         ];
 
+        $xmlFieldsToRemove = [];
+
+        if (version_compare(Version::VERSION, '5.5.0-b1', '<')) {
+            $xmlFieldsToRemove = ['Referrers_visitorsFromAIAssistants', 'Referrers_distinctAIAssistants', 'Referrers_visitorsFromAIAssistants_percent'];
+        }
+
         $apiToTest   = [];
         foreach ($apis as $api) {
             $apiToTest[] = [
@@ -94,6 +101,7 @@ public function getApiForTesting()
                     // when calling CustomVariables.getUsagesOfSlots, new archives are created until 'today',
                     // which increments idsubdatatable, but we need to have deterministic idsubdatatable
                     'apiNotToCall' => ['CustomVariables.getUsagesOfSlots'],
+                    'xmlFieldsToRemove' => $xmlFieldsToRemove,
                 ]
             ];
         }
diff --git a/tests/System/expected/test___API.get_year.xml b/tests/System/expected/test___API.get_year.xml
index 8eab3ab..6fec697 100644
--- a/tests/System/expected/test___API.get_year.xml
+++ b/tests/System/expected/test___API.get_year.xml
@@ -17,11 +17,13 @@
 	<avg_time_on_site_returning>0</avg_time_on_site_returning>
 	<Referrers_visitorsFromSearchEngines>0</Referrers_visitorsFromSearchEngines>
 	<Referrers_visitorsFromSocialNetworks>0</Referrers_visitorsFromSocialNetworks>
+	<Referrers_visitorsFromAIAssistants>0</Referrers_visitorsFromAIAssistants>
 	<Referrers_visitorsFromDirectEntry>1</Referrers_visitorsFromDirectEntry>
 	<Referrers_visitorsFromWebsites>0</Referrers_visitorsFromWebsites>
 	<Referrers_visitorsFromCampaigns>0</Referrers_visitorsFromCampaigns>
 	<Referrers_distinctSearchEngines>0</Referrers_distinctSearchEngines>
 	<Referrers_distinctSocialNetworks>0</Referrers_distinctSocialNetworks>
+	<Referrers_distinctAIAssistants>0</Referrers_distinctAIAssistants>
 	<Referrers_distinctKeywords>0</Referrers_distinctKeywords>
 	<Referrers_distinctWebsites>0</Referrers_distinctWebsites>
 	<Referrers_distinctWebsitesUrls>0</Referrers_distinctWebsitesUrls>
@@ -69,6 +71,7 @@
 	<nb_keywords>0</nb_keywords>
 	<Referrers_visitorsFromDirectEntry_percent>100%</Referrers_visitorsFromDirectEntry_percent>
 	<Referrers_visitorsFromSearchEngines_percent>0%</Referrers_visitorsFromSearchEngines_percent>
+	<Referrers_visitorsFromAIAssistants_percent>0%</Referrers_visitorsFromAIAssistants_percent>
 	<Referrers_visitorsFromCampaigns_percent>0%</Referrers_visitorsFromCampaigns_percent>
 	<Referrers_visitorsFromSocialNetworks_percent>0%</Referrers_visitorsFromSocialNetworks_percent>
 	<Referrers_visitorsFromWebsites_percent>0%</Referrers_visitorsFromWebsites_percent>