diff --git a/pylint_sqlalchemy/dml_no_argument.py b/pylint_sqlalchemy/dml_no_argument.py
index 48f5e32..29fa986 100644
--- a/pylint_sqlalchemy/dml_no_argument.py
+++ b/pylint_sqlalchemy/dml_no_argument.py
@@ -1,6 +1,9 @@
 """Linter to fix missing `dml` argument for pylint and sqlalchemy."""
 import astroid
-import astroid.node_classes
+try:
+    import astroid.node_classes
+except ModuleNotFoundError:
+    import astroid.nodes.node_classes
 
 
 def strip_dml(node):