diff --git a/conditional_fields.module b/conditional_fields.module
index 59dbcd6..fc98713 100644
--- a/conditional_fields.module
+++ b/conditional_fields.module
@@ -180,8 +180,8 @@ function conditional_fields_element_after_build($element, &$form_state) {
   $build_info = $form_state->getBuildInfo();
   if (method_exists($build_info['callback_object'], 'getEntity')) {
     $entity = $build_info['callback_object']->getEntity();
-    if (method_exists($entity, 'getType')) {
-      $bundle = $entity->getType();
+    if (method_exists($entity, 'bundle')) {
+      $bundle = $entity->bundle();
       $entity_type = $entity->getEntityTypeId();
 
       $dependencies = conditional_fields_load_dependencies($entity_type, $bundle);
-- 
2.7.4