diff --git a/src/ActivityRecordStorage.php b/src/ActivityRecordStorage.php
index 312ba64..84521cc 100644
--- a/src/ActivityRecordStorage.php
+++ b/src/ActivityRecordStorage.php
@@ -2,7 +2,7 @@
 
 namespace Drupal\entity_activity_tracker;
 
-use Drupal\Core\Database\Driver\mysql\Connection;
+use Drupal\Core\Database\Connection;
 use Drupal\Core\Entity\ContentEntityInterface;
 use Drupal\Component\Datetime\TimeInterface;
 use Drupal\Core\Database\Query\SelectInterface;
@@ -14,9 +14,9 @@
 class ActivityRecordStorage implements ActivityRecordStorageInterface {
 
   /**
-   * Drupal\Core\Database\Driver\mysql\Connection definition.
+   * Database connection.
    *
-   * @var \Drupal\Core\Database\Driver\mysql\Connection
+   * @var \Drupal\Core\Database\Connection
    */
   protected $database;
 
@@ -37,7 +37,7 @@ class ActivityRecordStorage implements ActivityRecordStorageInterface {
   /**
    * Constructs a new ActivityRecordStorage object.
    *
-   * @param \Drupal\Core\Database\Driver\mysql\Connection $database
+   * @param \Drupal\Core\Database\Connection $database
    *   The active database connection.
    * @param \Drupal\Component\Datetime\TimeInterface $date_time
    *   The time service.
diff --git a/src/Controller/ExportActivityRecordsController.php b/src/Controller/ExportActivityRecordsController.php
index e2ad4af..8d899b5 100644
--- a/src/Controller/ExportActivityRecordsController.php
+++ b/src/Controller/ExportActivityRecordsController.php
@@ -4,7 +4,7 @@
 
 use Drupal\Core\Config\ImmutableConfig;
 use Drupal\Core\Controller\ControllerBase;
-use Drupal\Core\Database\Driver\mysql\Connection;
+use Drupal\Core\Database\Connection;
 use Symfony\Component\HttpFoundation\Response;
 use Symfony\Component\DependencyInjection\ContainerInterface;
 use Symfony\Component\HttpFoundation\ResponseHeaderBag;
@@ -15,9 +15,9 @@
 class ExportActivityRecordsController extends ControllerBase {
 
   /**
-   * Drupal\Core\Database\Driver\mysql\Connection definition.
+   * Database connection.
    *
-   * @var \Drupal\Core\Database\Driver\mysql\Connection
+   * @var \Drupal\Core\Database\Connection
    */
   protected $database;
 
@@ -31,7 +31,7 @@ class ExportActivityRecordsController extends ControllerBase {
   /**
    * Constructs a new GroupMembershipController.
    *
-   * @param \Drupal\Core\Database\Driver\mysql\Connection $database
+   * @param \Drupal\Core\Database\Connection $database
    *   The active database connection.
    * @param \Drupal\Core\Config\ImmutableConfig $filesystem_settings
    *   The 'system.file' config.