From 57c88d5fd6a2a6b2c449c4290da933f6eb3cd0a7 Mon Sep 17 00:00:00 2001
From: Esben Sparre Andreasen <esbena@github.com>
Date: Wed, 30 Mar 2022 23:09:44 +0200
Subject: [PATCH] improve access path strings

---
 .../experimental/adaptivethreatmodeling/EndpointFeatures.qll  | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/javascript/ql/experimental/adaptivethreatmodeling/lib/experimental/adaptivethreatmodeling/EndpointFeatures.qll b/javascript/ql/experimental/adaptivethreatmodeling/lib/experimental/adaptivethreatmodeling/EndpointFeatures.qll
index 3436c3871f2..cc300c749ad 100644
--- a/javascript/ql/experimental/adaptivethreatmodeling/lib/experimental/adaptivethreatmodeling/EndpointFeatures.qll
+++ b/javascript/ql/experimental/adaptivethreatmodeling/lib/experimental/adaptivethreatmodeling/EndpointFeatures.qll
@@ -473,7 +473,7 @@ private module SyntacticUtilities {
             then result = "import(" + getSimpleImportPath(e) + ")"
             else
               if e instanceof AwaitExpr
-              then result = getSimpleAccessPath(e.(AwaitExpr).getOperand().flow()) + ".then()"
+              then result = "(await " + getSimpleAccessPath(e.(AwaitExpr).getOperand().flow()) + ")"
               else
                 if node instanceof DataFlow::PropRead
                 then
@@ -492,7 +492,7 @@ private module SyntacticUtilities {
     if exists(i.getImportedPath().getValue())
     then
       exists(string p | p = i.getImportedPath().getValue() |
-        if p.matches(".%") then result = p else result = "!" // hide absolute imports from the ML training
+        if p.matches(".%") then result = "\"p\"" else result = "!" // hide absolute imports from the ML training
       )
     else result = "?"
   }