diff --git a/autoapi/inheritance_diagrams.py b/autoapi/inheritance_diagrams.py
index 67ef8a62..297ba0c0 100644
--- a/autoapi/inheritance_diagrams.py
+++ b/autoapi/inheritance_diagrams.py
@@ -43,7 +43,8 @@ def _import_class(name, currmodule):
 
     if isinstance(target, astroid.Module):
         return [
-            child for child in target.get_children()
+            child
+            for child in target.get_children()
             if isinstance(child, astroid.ClassDef)
         ]