diff --git a/pygolf/helper/walker.py b/pygolf/helper/walker.py
index f5f7e26..cbbc7b5 100644
--- a/pygolf/helper/walker.py
+++ b/pygolf/helper/walker.py
@@ -1,4 +1,4 @@
-from astroid.node_classes import NodeNG
+from astroid.nodes import NodeNG
 
 
 def walk(node: NodeNG):
diff --git a/pygolf/optimization_phases/always_apply_phase.py b/pygolf/optimization_phases/always_apply_phase.py
index cd1d2f4..943dae4 100644
--- a/pygolf/optimization_phases/always_apply_phase.py
+++ b/pygolf/optimization_phases/always_apply_phase.py
@@ -1,6 +1,6 @@
 from typing import Iterator
 
-from astroid.node_classes import NodeNG
+from astroid.nodes import NodeNG
 
 from pygolf.rules import *
 
diff --git a/pygolf/optimization_phases/phase.py b/pygolf/optimization_phases/phase.py
index dd77821..32c0b26 100644
--- a/pygolf/optimization_phases/phase.py
+++ b/pygolf/optimization_phases/phase.py
@@ -1,7 +1,7 @@
 import abc
 from typing import Iterator
 
-from astroid.node_classes import NodeNG
+from astroid.nodes import NodeNG
 
 from pygolf.rules import AstroidRule
 
diff --git a/pygolf/optimization_phases/rename_phase.py b/pygolf/optimization_phases/rename_phase.py
index 3d1f894..6866539 100644
--- a/pygolf/optimization_phases/rename_phase.py
+++ b/pygolf/optimization_phases/rename_phase.py
@@ -1,6 +1,6 @@
 from typing import Iterator
 
-from astroid.node_classes import NodeNG
+from astroid.nodes import NodeNG
 
 from pygolf.name_finder import NameFinder
 from pygolf.optimizers.batch_optimizer import BatchOptimizer
diff --git a/pygolf/optimizers/batch_optimizer.py b/pygolf/optimizers/batch_optimizer.py
index 7ef1faa..32b92a6 100644
--- a/pygolf/optimizers/batch_optimizer.py
+++ b/pygolf/optimizers/batch_optimizer.py
@@ -1,6 +1,6 @@
 from typing import *
 
-from astroid.node_classes import NodeNG
+from astroid.nodes import NodeNG
 
 from pygolf.helper import walker
 from pygolf.name_finder import NameFinder
diff --git a/pygolf/optimizers/optimizer.py b/pygolf/optimizers/optimizer.py
index 2393a77..73575a9 100644
--- a/pygolf/optimizers/optimizer.py
+++ b/pygolf/optimizers/optimizer.py
@@ -1,7 +1,7 @@
 import abc
 from typing import Iterator
 
-from astroid.node_classes import NodeNG
+from astroid.nodes import NodeNG
 
 
 class Optimizer(metaclass=abc.ABCMeta):
diff --git a/pygolf/pygolfer.py b/pygolf/pygolfer.py
index 05e9071..36b3bea 100644
--- a/pygolf/pygolfer.py
+++ b/pygolf/pygolfer.py
@@ -1,7 +1,7 @@
 from typing import List
 
 import astroid as ast
-from astroid.node_classes import NodeNG
+from astroid.nodes import NodeNG
 
 from pygolf.optimization_phases import all_phases
 from pygolf.rules import AstroidRule
diff --git a/pygolf/rules/astroid_rule.py b/pygolf/rules/astroid_rule.py
index e86e7f1..56d01ad 100644
--- a/pygolf/rules/astroid_rule.py
+++ b/pygolf/rules/astroid_rule.py
@@ -1,6 +1,6 @@
 import abc
 
-from astroid.node_classes import NodeNG
+from astroid.nodes import NodeNG
 
 from pygolf.rules.version import Version
 
diff --git a/pygolf/unparser.py b/pygolf/unparser.py
index d19fd64..645c4ec 100644
--- a/pygolf/unparser.py
+++ b/pygolf/unparser.py
@@ -1,7 +1,7 @@
 from typing import *
 
 import astroid as ast
-from astroid.node_classes import NodeNG
+from astroid.nodes import NodeNG
 
 from pygolf.errors.python_2_code_detected import Python2CodeDetected