From 48cb075b3be7648e41fac929acac49d6dde52ca1 Mon Sep 17 00:00:00 2001
From: Rasmus Wriedt Larsen <rasmuswl@github.com>
Date: Mon, 18 Sep 2023 14:25:04 +0200
Subject: [PATCH] Python: More import fixes

:thinkies: turns out that .getASubclass*() had to be applied everywhere...
---
 .../lib/semmle/python/frameworks/internal/SubclassFinder.qll   | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/python/ql/lib/semmle/python/frameworks/internal/SubclassFinder.qll b/python/ql/lib/semmle/python/frameworks/internal/SubclassFinder.qll
index 36626677dc1..1861eeea6ab 100644
--- a/python/ql/lib/semmle/python/frameworks/internal/SubclassFinder.qll
+++ b/python/ql/lib/semmle/python/frameworks/internal/SubclassFinder.qll
@@ -162,7 +162,8 @@ module NotExposed {
     FindSubclassesSpec spec, string newAliasFullyQualified, ImportMember importMember, Module mod,
     Location loc
   ) {
-    importMember = newOrExistingModeling(spec).getAValueReachableFromSource().asExpr() and
+    importMember =
+      newOrExistingModeling(spec).getASubclass*().getAValueReachableFromSource().asExpr() and
     importMember.getScope() = mod and
     loc = importMember.getLocation() and
     (