diff --git a/src/Plugin/Block/CustomersAlsoBoughtBlock.php b/src/Plugin/Block/CustomersAlsoBoughtBlock.php
index 1986aab..802a857 100644
--- a/src/Plugin/Block/CustomersAlsoBoughtBlock.php
+++ b/src/Plugin/Block/CustomersAlsoBoughtBlock.php
@@ -177,7 +177,7 @@ class CustomersAlsoBoughtBlock extends BlockBase implements ContainerFactoryPlug

     $config = $this->getConfiguration();

-    $view_modes = $this->entityDisplayRepository->getViewModes('node');
+    $view_modes = $this->entityDisplayRepository->getViewModes('commerce_product');
     $modes = [];
     foreach ($view_modes as $key => $view_mode) {
       $modes[$key] = $view_mode['label'];