diff --git a/gadget-code/src/lib/code-session.ts b/gadget-code/src/lib/code-session.ts
index f04fffe..09cca05 100644
--- a/gadget-code/src/lib/code-session.ts
+++ b/gadget-code/src/lib/code-session.ts
@@ -6,7 +6,7 @@ import {
   GadgetSocket,
   SocketSession,
   SocketSessionType,
-} from "./socket-session";
+} from "./socket-session.js";
 import {
   IChatSession,
   IDroneRegistration,
diff --git a/gadget-code/src/lib/drone-session.ts b/gadget-code/src/lib/drone-session.ts
index b1bfdee..e700383 100644
--- a/gadget-code/src/lib/drone-session.ts
+++ b/gadget-code/src/lib/drone-session.ts
@@ -12,9 +12,9 @@ import {
   GadgetSocket,
   SocketSession,
   SocketSessionType,
-} from "./socket-session";
-import SocketService from "../services/socket";
-import { ChatTurn } from "../models/chat-turn";
+} from "./socket-session.js";
+import { SocketService } from "../services/index.js";
+import { ChatTurn } from "../models/chat-turn.js";
 
 export class DroneSession extends SocketSession {
   protected type: SocketSessionType = SocketSessionType.Drone;
diff --git a/gadget-code/src/lib/socket-session.ts b/gadget-code/src/lib/socket-session.ts
index e0243b5..77b6bdb 100644
--- a/gadget-code/src/lib/socket-session.ts
+++ b/gadget-code/src/lib/socket-session.ts
@@ -9,7 +9,7 @@ import {
   ServerToClientEvents,
   SocketData,
 } from "@gadget/api";
-import { DtpLog } from "./log";
+import { DtpLog } from "./log.js";
 
 export enum SocketSessionType {
   Code = "code",