diff --git a/config_update_ui/src/ConfigUpdateUiCliService.php b/config_update_ui/src/ConfigUpdateUiCliService.php
index 34bef19..6b5d7bb 100644
--- a/config_update_ui/src/ConfigUpdateUiCliService.php
+++ b/config_update_ui/src/ConfigUpdateUiCliService.php
@@ -5,6 +5,7 @@ namespace Drupal\config_update_ui;
 use Drupal\config_update\ConfigDiffer;
 use Drupal\config_update\ConfigListerWithProviders;
 use Drupal\config_update\ConfigReverter;
+use Drupal\config_ui\ConfigUpdateUiDrush8Logger;
 use Drupal\Component\Diff\DiffFormatter;
 use Drupal\Core\Entity\EntityTypeManagerInterface;
 use Consolidation\OutputFormatters\StructuredData\RowsOfFields;
@@ -313,9 +314,9 @@ class ConfigUpdateUiCliService {
    * Registers a logger.
    *
    * @param \Psr\Log\LoggerInterface $logger
-   *   The logging object to use. For Drush 8, you must use an object of class
-   *   \Drupal\config_ui\ConfigUpdateUiDrush8Logger. For Drush 9 and later,
-   *   use the output of \Drush\Drush::logger().
+   *   The logging object to use. For Drush 8, use an object of class
+   *   \Drupal\config_ui\ConfigUpdateUiDrush8Logger (or a subclass). For Drush
+   *   9 and later, use the output of \Drush\Drush::logger().
    */
   public function setLogger(LoggerInterface $logger) {
     $this->logger = $logger;
@@ -404,7 +405,7 @@ class ConfigUpdateUiCliService {
    *   data object of rows strings.
    */
   protected function outputRows(array $output) {
-    if (get_class($this->logger) === 'ConfigUpdateUiDrush8Logger') {
+    if ($this->logger instanceof ConfigUpdateUiDrush8Logger) {
       return $output;
     }
     else {