diff --git a/agent.js b/agent.js
index 161e181..b3471a5 100644
--- a/agent.js
+++ b/agent.js
@@ -1,10 +1,10 @@
 import { createAgent } from 'langchain'
 import { MemorySaver } from '@langchain/langgraph'
 import { randomUUIDv7 } from 'bun'
-import { llm } from './llm'
-import { web_search_tool } from './tools'
+import { llm } from './llm.js'
+import { web_search_tool } from './tools.js'
 import ora from 'ora'
-import { ui } from './ui'
+import { ui } from './ui.js'
 
 const checkpointer = new MemorySaver()
 
diff --git a/bro b/bro
index 76581f2..cb6666c 100755
--- a/bro
+++ b/bro
@@ -3,7 +3,7 @@ import { rl } from './io.js'
 import { handle_args } from './nonInteractive.js'
 import { call_agent } from './agent.js'
 import { end, setup } from './setup.js'
-import { ui } from './ui'
+import { ui } from './ui.js'
 
 setup()
 
diff --git a/nonInteractive.js b/nonInteractive.js
index 5e411fc..203ce82 100644
--- a/nonInteractive.js
+++ b/nonInteractive.js
@@ -1,5 +1,5 @@
-import { call_agent } from './agent'
-import { ui } from './ui'
+import { call_agent } from './agent.js'
+import { ui } from './ui.js'
 
 /**
  * Handles non-interactive mode with streaming and error feedback.
diff --git a/setup.js b/setup.js
index 745828d..8e39c99 100644
--- a/setup.js
+++ b/setup.js
@@ -1,5 +1,5 @@
-import { rl } from './io'
-import { ui } from './ui'
+import { rl } from './io.js'
+import { ui } from './ui.js'
 
 /**
  * Sets up the CLI environment, including signal handlers and help flag check.