diff --git a/src/Tests/Update/DerUpdateTest.php b/src/Tests/Update/DerUpdateTest.php
index 3eae178..4c93675 100644
--- a/src/Tests/Update/DerUpdateTest.php
+++ b/src/Tests/Update/DerUpdateTest.php
@@ -35,13 +35,19 @@ class DerUpdateTest extends UpdatePathTestBase {
    * @see dynamic_entity_reference_update_8001()
    */
   public function testUpdate8001() {
+    $connection = \Drupal::database();
+    if ($connection->driver() == 'mysql') {
+      // This might forcer an 1071 Specified key was too long; max key length
+      // is 767 bytes error if innodb_large_prefix is ON so test it.
+      $connection->query('ALTER TABLE {entity_test__field_test} ROW_FORMAT=compact');
+    }
     $this->runUpdates();
     // The db dump contain two entity_test entities referencing one entity_test
     // entity and one entity_test_mul entity.
-    $this->assertEqual([1, 1, 1, 1], \Drupal::database()->query('SELECT field_test_target_id_int FROM {entity_test__field_test}')->fetchCol());
+    $this->assertEqual([1, 1, 1, 1], $connection->query('SELECT field_test_target_id_int FROM {entity_test__field_test}')->fetchCol());
     // The db dump contain two entity_test_mul entities referencing one
     // entity_test entity and a entity_test_mul entity.
-    $this->assertEqual([1, 1, 1, 1], \Drupal::database()->query('SELECT field_test_mul_target_id_int FROM {entity_test_mul__field_test_mul}')->fetchCol());
+    $this->assertEqual([1, 1, 1, 1], $connection->query('SELECT field_test_mul_target_id_int FROM {entity_test_mul__field_test_mul}')->fetchCol());
     $referenced_entity = EntityTestStringId::create([
       'id' => 'test',
     ]);
@@ -49,8 +55,8 @@ class DerUpdateTest extends UpdatePathTestBase {
     $entity = EntityTestMul::load(3);
     $entity->field_test_mul[] = $referenced_entity;
     $entity->save();
-    $this->assertEqual([1, 1, 1, 1, 0], \Drupal::database()->query('SELECT field_test_mul_target_id_int FROM {entity_test_mul__field_test_mul} ORDER BY entity_id, delta')->fetchCol());
-    $this->assertEqual([1, 1, 1, 1, 'test'], \Drupal::database()->query('SELECT field_test_mul_target_id FROM {entity_test_mul__field_test_mul} ORDER BY entity_id, delta')->fetchCol());
+    $this->assertEqual([1, 1, 1, 1, 0], $connection->query('SELECT field_test_mul_target_id_int FROM {entity_test_mul__field_test_mul} ORDER BY entity_id, delta')->fetchCol());
+    $this->assertEqual([1, 1, 1, 1, 'test'], $connection->query('SELECT field_test_mul_target_id FROM {entity_test_mul__field_test_mul} ORDER BY entity_id, delta')->fetchCol());
   }
 
 }