diff --git a/autoapi/inheritance_diagrams.py b/autoapi/inheritance_diagrams.py
index 0799acd..5740fa7 100644
--- a/autoapi/inheritance_diagrams.py
+++ b/autoapi/inheritance_diagrams.py
@@ -52,7 +52,7 @@ def _import_class(name, currmodule):
 
     if isinstance(target, astroid.Module):
         classes = []
-        for child in target.children:
+        for child in target.get_children():
             if isinstance(child, astroid.ClassDef):
                 classes.append(child)
         return classes