diff --git a/auto_menutitle.info.yml b/auto_menutitle.info.yml
index 872aaaa..22ceddf 100644
--- a/auto_menutitle.info.yml
+++ b/auto_menutitle.info.yml
@@ -2,5 +2,6 @@ name: Automatic Menu Titles
 type: module
 description: "Allows content types to set the default value of the 'Provide a menu link' checkbox."
 core: 8.x
+core_version_requirement: ^8 || ^9
 dependencies:
   - menu_ui:menu_ui
diff --git a/auto_menutitle.install b/auto_menutitle.install
index 9511a48..39947e4 100644
--- a/auto_menutitle.install
+++ b/auto_menutitle.install
@@ -1,5 +1,5 @@
 <?php
-
+use Drupal\node\Entity\NodeType;
 /**
  * @file
  * Install, update and uninstall functions for the auto_menutitle module.
@@ -19,7 +19,7 @@ function auto_menutitle_install() {
  */
 function auto_menutitle_uninstall() {
   // Delete variables.
-  $node_types = array_keys(node_type_get_types());
+  $node_types = array_keys(NodeType::loadMultiple());
   foreach ($node_types as $node_type) {
     \Drupal::state()->delete('auto_menutitle_' . $node_type);
   }