diff --git a/tests/unit/Flash/Direct/Helper/FlashBase.php b/tests/unit/Flash/Direct/Helper/FlashBase.php
index df3956d5287..9ea2e640a86 100644
--- a/tests/unit/Flash/Direct/Helper/FlashBase.php
+++ b/tests/unit/Flash/Direct/Helper/FlashBase.php
@@ -62,7 +62,7 @@ public function testFlashDirectImplicitFlushHtml(UnitTester $I)
     private function stringTest(UnitTester $I, string $function)
     {
         $flash = new Direct(new Escaper());
-        $flash->setClasses($this->classes);
+        $flash->setCssClasses($this->classes);
 
         $message = 'sample message';
 
@@ -213,7 +213,7 @@ public function testFlashDirectNoImplicitFlushNoHtml(UnitTester $I)
     public function testFlashDirectWithAutoEscaping(UnitTester $I)
     {
         $flash = new Direct(new Escaper());
-        $flash->setClasses($this->classes);
+        $flash->setCssClasses($this->classes);
 
         $flash->setAutomaticHtml(false);
         $flash->setImplicitFlush(false);
diff --git a/tests/unit/Flash/DirectCest.php b/tests/unit/Flash/DirectCest.php
index 36d3db69c2e..0ca1561a604 100644
--- a/tests/unit/Flash/DirectCest.php
+++ b/tests/unit/Flash/DirectCest.php
@@ -103,7 +103,7 @@ protected function getFlash()
 
         $flash = new Direct();
         $flash->setDI($container);
-        $flash->setClasses($this->classes);
+        $flash->setCssClasses($this->classes);
 
         return $flash;
     }
diff --git a/tests/unit/Flash/Session/OutputCest.php b/tests/unit/Flash/Session/OutputCest.php
index 6d70acfa027..b0925747a75 100644
--- a/tests/unit/Flash/Session/OutputCest.php
+++ b/tests/unit/Flash/Session/OutputCest.php
@@ -79,7 +79,7 @@ protected function getFlash()
         $container = $this->getDi();
         $flash     = new Session();
         $flash->setDI($container);
-        $flash->setClasses($this->classes);
+        $flash->setCssClasses($this->classes);
 
         return $flash;
     }
diff --git a/tests/unit/Flash/Session/SetCustomTemplateCest.php b/tests/unit/Flash/Session/SetCustomTemplateCest.php
index ddfee9675fc..e15d9b9e401 100644
--- a/tests/unit/Flash/Session/SetCustomTemplateCest.php
+++ b/tests/unit/Flash/Session/SetCustomTemplateCest.php
@@ -46,7 +46,7 @@ protected function getFlash()
 
         $flash     = new Session();
         $flash->setDI($container);
-        $flash->setClasses($this->classes);
+        $flash->setCssClasses($this->classes);
 
         return $flash;
     }
diff --git a/tests/unit/Flash/SessionCest.php b/tests/unit/Flash/SessionCest.php
index bf3d6f30e9f..224098e9a67 100644
--- a/tests/unit/Flash/SessionCest.php
+++ b/tests/unit/Flash/SessionCest.php
@@ -142,7 +142,7 @@ protected function getFlash()
 
         $flash = new Session();
         $flash->setDI($container);
-        $flash->setClasses($this->classes);
+        $flash->setCssClasses($this->classes);
 
         return $flash;
     }