Socket failing to transmit/receive when using 'upgrade' event and socket.write on Expressjs server

import { WebSocketServer } from "ws";
import express from "express";

const app = express();

app.use((rq, rs) => {
    rs.sendStatus(404);
});

const wss = new WebSocketServer({ noServer: true, path: "/" });
wss.on('connection', socket => {
    socket.send("Hello World!");
    socket.on('message', data => {
        console.log(`Got data: ${data.toString()}`);
        
        if (data.toString() === "close") {
            socket.send("Bye!");
            socket.close();
        }
    });
});

const server = app.listen(3000);

server.on('upgrade', (req, socket, head) => {

    console.log('upgrade request');
    socket.write("HTTP/1.1 404 Not Found\r\n\r\n");
    socket.end();
    wss.handleUpgrade(req, socket, head, ws => {
        wss.emit('connection', ws, req);
    });
});