diff --git a/src/EventSubscriber/MigrateProduct.php b/src/EventSubscriber/MigrateProduct.php
index 800d129..2a934f3 100644
--- a/src/EventSubscriber/MigrateProduct.php
+++ b/src/EventSubscriber/MigrateProduct.php
@@ -35,17 +35,10 @@ class MigrateProduct implements EventSubscriberInterface {
     // Ensure the back reference to the product is correct for all variations
     // of this product.
     if ($event->getMigration()->id() == 'd7_commerce_product') {
-      $variations = $event->getRow()->getDestinationProperty('variations');
       $product_id = $event->getRow()->getDestinationProperty('product_id');
-      $product = Product::load($product_id);
-      if ($product) {
-        foreach ($variations as $variation) {
-          $variation = ProductVariation::load($variation);
-          if ($variation) {
-            $variation->set('product_id', $product_id);
-            $variation->save();
-          }
-        }
+      if ($product = Product::load($product_id)) {
+        // Save updates the back-reference on each product variation.
+        $product->save();
       }
     }
   }