diff --git a/internal/jujuapi/websocket.go b/internal/jujuapi/websocket.go
index bd529648f..bca6c387d 100644
--- a/internal/jujuapi/websocket.go
+++ b/internal/jujuapi/websocket.go
@@ -55,7 +55,7 @@ func (s *apiServer) ServeWS(ctx context.Context, conn *websocket.Conn) {
 	controllerRoot := newControllerRoot(s.jimm, s.params, identityId)
 	s.cleanup = controllerRoot.cleanup
 	Dblogger := controllerRoot.newAuditLogger()
-	serveRoot(context.Background(), controllerRoot, Dblogger, conn)
+	serveRoot(ctx, controllerRoot, Dblogger, conn)
 }
 
 // Kill implements the rpc.Killer interface.