diff --git a/webform.install b/webform.install
index 52a79fa..0101df5 100644
--- a/webform.install
+++ b/webform.install
@@ -739,7 +739,7 @@ function webform_uninstall() {
     ->range(0, 1)
     ->execute();
   $entity_manager = \Drupal::entityManager();
-  $additional_fields = array_diff_key($entity_manager->getFieldDefinitions('node', 'webform'), $entity_manager->getBaseFieldDefinitions('node', 'webform'));
+  $additional_fields = array_diff_key($entity_manager->getFieldDefinitions('node', 'webform'), $entity_manager->getBaseFieldDefinitions('node'));
   unset($additional_fields['body']);
   if (!$webform_nodes && !$additional_fields) {
     $node_type = NodeType::load('webform');