Python: Minor simplification of ActiveThreatModelSource

Co-authored-by: Taus <[email protected]>
github · Sep 23, 2024 · 535db98 · 535db98
1 parent 4a21a85
commit 535db98
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/ql/lib/semmle/python/Concepts.qll b/python/ql/lib/semmle/python/Concepts.qll
@@ -58,11 +58,11 @@ module ThreatModelSource {
 /**
  * A data flow source that is enabled in the current threat model configuration.
  */
-class ActiveThreatModelSource extends DataFlow::Node {
+class ActiveThreatModelSource extends ThreatModelSource {
   ActiveThreatModelSource() {
     exists(string kind |
       currentThreatModel(kind) and
-      this.(ThreatModelSource).getThreatModel() = kind
+      this.getThreatModel() = kind
     )
   }
 }