diff --git a/contract-tests/sdkClientEntity.js b/contract-tests/sdkClientEntity.js
index 3830730d4..bd9ad2d4f 100644
--- a/contract-tests/sdkClientEntity.js
+++ b/contract-tests/sdkClientEntity.js
@@ -1,12 +1,11 @@
 import got from 'got';
 import ld, {
   LDConcurrentExecution,
-  LDExecution,
   LDExecutionOrdering,
   LDMigrationError,
   LDMigrationSuccess,
   LDSerialExecution,
-  Migration,
+  createMigration,
 } from 'node-server-sdk';
 
 import BigSegmentTestStore from './BigSegmentTestStore.js';
@@ -172,7 +171,7 @@ export async function newSdkClientEntity(options) {
         const migrationOperation = params.migrationOperation;
         const readExecutionOrder = migrationOperation.readExecutionOrder;
 
-        const migration = new Migration(client, {
+        const migration = createMigration(client, {
           execution: getExecution(readExecutionOrder),
           latencyTracking: migrationOperation.trackLatency,
           errorTracking: migrationOperation.trackErrors,