diff --git a/composer.json b/composer.json
index 574504315711d9b1bf32b90668cafc6bbdc80aad..a8c452200ef68021b32328b9b6391333597ee93b 100644
--- a/composer.json
+++ b/composer.json
@@ -40,7 +40,7 @@
     "drupal/key": "^1.14",
     "firebase/php-jwt": "^5.0",
     "lusitanian/oauth": "^0.8.11",
-    "drupal/dynamic_entity_reference": ">=1.9 <3.0",
+    "drupal/dynamic_entity_reference": "^1.9 || ^2.0",
     "drupal/typed_data": "^1.0-alpha5",
     "messageagency/force.com-toolkit-for-php": "^1.0.1",
     "ext-soap": "*"
diff --git a/modules/salesforce_jwt/tests/src/FunctionalJavascript/SalesforceJwtTest.php b/modules/salesforce_jwt/tests/src/FunctionalJavascript/SalesforceJwtTest.php
index 10e00ef3e56e34fbee017b981b66cf3139511fa2..6ca84505106bbf0ae3ce7a75df3d7dc5c2ca7716 100644
--- a/modules/salesforce_jwt/tests/src/FunctionalJavascript/SalesforceJwtTest.php
+++ b/modules/salesforce_jwt/tests/src/FunctionalJavascript/SalesforceJwtTest.php
@@ -3,8 +3,6 @@
 namespace Drupal\Tests\salesforce_jwt\FunctionalJavascript;
 
 use Drupal\FunctionalJavascriptTests\WebDriverTestBase;
-use Drupal\key\Entity\Key;
-use Drupal\Tests\key\Functional\KeyTestTrait;
 
 /**
  * Test JWT Auth.
@@ -20,7 +18,12 @@ class SalesforceJwtTest extends WebDriverTestBase {
    */
   protected $defaultTheme  = 'stark';
 
-  use KeyTestTrait;
+  /**
+   * A key entity to use for testing.
+   *
+   * @var \Drupal\key\KeyInterface
+   */
+  protected $testKey;
 
   /**
    * Modules.
@@ -56,7 +59,7 @@ class SalesforceJwtTest extends WebDriverTestBase {
     $this->adminUser = $this->drupalCreateUser(['authorize salesforce']);
     $this->drupalLogin($this->adminUser);
     $this->createTestKey(self::KEY_ID, 'authentication', 'file');
-    Key::load(self::KEY_ID)
+    \Drupal\key\Entity\Key::load(self::KEY_ID)
       ->set('key_provider_settings', [
         'file_location' => __DIR__ . '/testKey.pem',
         'strip_line_breaks' => FALSE,
@@ -104,4 +107,23 @@ class SalesforceJwtTest extends WebDriverTestBase {
     $assert_session->pageTextContainsOnce('Salesforce JWT OAuth');
   }
 
+  /**
+   * Make a key for testing operations that require a key.
+   */
+  protected function createTestKey($id, $type = NULL, $provider = NULL) {
+    $keyArgs = [
+      'id' => $id,
+      'label' => 'Test key',
+    ];
+    if ($type != NULL) {
+      $keyArgs['key_type'] = $type;
+    }
+    if ($provider != NULL) {
+      $keyArgs['key_provider'] = $provider;
+    }
+    $this->testKey = \Drupal\key\Entity\Key::create($keyArgs);
+    $this->testKey->save();
+    return $this->testKey;
+  }
+
 }
diff --git a/modules/salesforce_mapping/composer.json b/modules/salesforce_mapping/composer.json
index c8c787a46c840c321e5e76228bf34633edb84ca2..8cdaac2ca15b347284a3eb697d28cbf1508e579a 100644
--- a/modules/salesforce_mapping/composer.json
+++ b/modules/salesforce_mapping/composer.json
@@ -10,7 +10,7 @@
   ],
   "require": {
     "php": ">=5.6.0",
-    "drupal/dynamic_entity_reference": "^2.0@alpha10",
+    "drupal/dynamic_entity_reference": "^1.9 || ^2.0",
     "drupal/typed_data": "^1.0-alpha5"
   },
   "extra": {
@@ -20,4 +20,4 @@
       }
     }
   }
-}
\ No newline at end of file
+}
diff --git a/modules/salesforce_mapping/salesforce_mapping.info.yml b/modules/salesforce_mapping/salesforce_mapping.info.yml
index 05888e777954300d239dfc0314dc24829bd02d3e..4295447841d6160c49f1f615f9c1af6b5f6c053a 100644
--- a/modules/salesforce_mapping/salesforce_mapping.info.yml
+++ b/modules/salesforce_mapping/salesforce_mapping.info.yml
@@ -5,5 +5,5 @@ package: Salesforce
 core_version_requirement: ^8.7.7 || ^9
 dependencies:
   - salesforce:salesforce
-  - dynamic_entity_reference:dynamic_entity_reference (>= 2.0-alpha10)
+  - dynamic_entity_reference:dynamic_entity_reference
   - typed_data:typed_data (>= 1.0-alpha5)