diff --git a/src/ResultPrinter/HTML.php b/src/ResultPrinter/HTML.php
index 1b0acb7..989df1a 100644
--- a/src/ResultPrinter/HTML.php
+++ b/src/ResultPrinter/HTML.php
@@ -297,7 +297,7 @@ protected function renderSubsteps(Meta $metaStep, $substepsBuffer)
     private function cleanMessage($exception)
     {
         $msg = $exception->getMessage();
-        if ($exception instanceof \PHPUnit\Framework\AssertionFailedError && $exception->getComparisonFailure()) {
+        if ($exception instanceof \PHPUnit\Framework\ExpectationFailedException && $exception->getComparisonFailure()) {
             $msg .= $exception->getComparisonFailure()->getDiff();
         }
         $msg = str_replace(['<info>','</info>','<bold>','</bold>'], ['','','',''], $msg);