diff --git a/src/IndexBuilder.php b/src/IndexBuilder.php
index 4945b95..33a461d 100644
--- a/src/IndexBuilder.php
+++ b/src/IndexBuilder.php
@@ -70,10 +70,10 @@ public function speedUpRefresh(Index $index): void
         $index->getSettings()->setRefreshInterval('1s');
     }
 
-    public function migrate(Index $currentIndex, array $params = []): Index
+    public function migrate(Index $currentIndex, array $params = [], array $context = []): Index
     {
         $pureIndexName = $this->indexNameMapper->getPureIndexName($currentIndex->getName());
-        $newIndex = $this->createIndex($pureIndexName);
+        $newIndex = $this->createIndex($pureIndexName, $context);
 
         $reindex = new Reindex($currentIndex, $newIndex, $params);
         $reindex->setWaitForCompletion(false);