diff --git a/src/Entity/ProductBundleItem.php b/src/Entity/ProductBundleItem.php
index 67b9b88..f36fe52 100644
--- a/src/Entity/ProductBundleItem.php
+++ b/src/Entity/ProductBundleItem.php
@@ -210,8 +210,8 @@ class ProductBundleItem extends ContentEntityBase implements BundleItemInterface
     // @todo: Figure out how to get the currently selected variation
     // without holding state in this object.
     // @see https://www.drupal.org/node/2831613
-    elseif (!$this->getDefaultVariation()->getUnitPrice()->isEmpty()) {
-      return $this->getDefaultVariation()->getUnitPrice()->first()->toPrice();
+    elseif (!$this->getDefaultVariation()->get('price')->isEmpty()) {
+      return $this->getDefaultVariation()->get('price')->first()->toPrice();
     }
   }