diff --git a/languages/tree-sitter-stack-graphs-typescript/src/stack-graphs.tsg b/languages/tree-sitter-stack-graphs-typescript/src/stack-graphs.tsg
index fa1cbf8f8..b7478b4e4 100644
--- a/languages/tree-sitter-stack-graphs-typescript/src/stack-graphs.tsg
+++ b/languages/tree-sitter-stack-graphs-typescript/src/stack-graphs.tsg
@@ -540,7 +540,8 @@ attribute node_symbol = node            => symbol = (source-text node), source_n
 
   ; module reference
   var mod_scope = mod_ref__ns
-  scan (path-normalize mod_path) {
+  ; normalize path and remove the extension as we want to match 'foo', 'foo.js', 'foo.ts', etc.
+  scan (path-normalize (replace mod_path "\.(js|ts|jsx|tsx)$" "")) {
     "([^/]+)/?" {
       node mod_ref
       attr (mod_ref) push_symbol = $1
diff --git a/languages/tree-sitter-stack-graphs-typescript/test/modules/import-extension-js.ts b/languages/tree-sitter-stack-graphs-typescript/test/modules/import-extension-js.ts
new file mode 100644
index 000000000..92817fbaf
--- /dev/null
+++ b/languages/tree-sitter-stack-graphs-typescript/test/modules/import-extension-js.ts
@@ -0,0 +1,6 @@
+/* --- path: src/foo.ts --- */
+export const bar = 42;
+
+/* --- path: src/index.ts --- */
+import { bar } from "./foo.js";
+//       ^ defined: 2
diff --git a/languages/tree-sitter-stack-graphs-typescript/test/modules/import-extension-none.ts b/languages/tree-sitter-stack-graphs-typescript/test/modules/import-extension-none.ts
new file mode 100644
index 000000000..f221684b4
--- /dev/null
+++ b/languages/tree-sitter-stack-graphs-typescript/test/modules/import-extension-none.ts
@@ -0,0 +1,6 @@
+/* --- path: src/foo.ts --- */
+export const bar = 42;
+
+/* --- path: src/index.ts --- */
+import { bar } from "./foo";
+//       ^ defined: 2
diff --git a/languages/tree-sitter-stack-graphs-typescript/test/modules/import-extension-ts.ts b/languages/tree-sitter-stack-graphs-typescript/test/modules/import-extension-ts.ts
new file mode 100644
index 000000000..1e46349c9
--- /dev/null
+++ b/languages/tree-sitter-stack-graphs-typescript/test/modules/import-extension-ts.ts
@@ -0,0 +1,6 @@
+/* --- path: src/foo.ts --- */
+export const bar = 42;
+
+/* --- path: src/index.ts --- */
+import { bar } from "./foo.ts";
+//       ^ defined: 2