diff --git a/tests/php/FileSubsitesTest.php b/tests/php/FileSubsitesTest.php
index 13b0ae7..57ab4f3 100644
--- a/tests/php/FileSubsitesTest.php
+++ b/tests/php/FileSubsitesTest.php
@@ -41,7 +41,7 @@ class FileSubsitesTest extends BaseSubsiteTest
         $this->assertEquals('FileTitle', $file->getTreeTitle());
         $this->assertTrue(singleton(Folder::class)->getCMSFields() instanceof FieldList);
         Subsite::changeSubsite(1);
-        $this->assertEquals($file->cacheKeyComponent(), 'subsite-1');
+        $this->assertEquals('subsite-1', $file->getExtensionInstance(FileSubsites::class)->cacheKeyComponent());
     }
 
     public function testWritingSubsiteID()