diff --git a/payment/payment.module b/payment/payment.module
index f732d4c..d39eba8 100644
--- a/payment/payment.module
+++ b/payment/payment.module
@@ -320,6 +320,21 @@ function payment_element_info() {
 }
 
 /**
+ * Implements hook_entity_bundle_info().
+ */
+function payment_entity_bundle_info() {
+  $bundles = array();
+  $manager = \Drupal::service('plugin.manager.payment.payment_method');
+  foreach ($manager->getDefinitions() as $plugin_id => $definition) {
+    $bundles['payment'][$plugin_id] = array(
+      'label' => $definition['label'],
+    );
+  }
+
+  return $bundles;
+}
+
+/**
  * Implements hook_ENTITY_TYPE_load().
  */
 function payment_payment_load(array $payments) {