diff --git a/cibuild/linkimports.sh b/cibuild/linkimports.sh
index 0f706f2ae..55a4ab882 100644
--- a/cibuild/linkimports.sh
+++ b/cibuild/linkimports.sh
@@ -59,6 +59,7 @@ _getenv
 _interactive_one
 _interactive_read
 _interactive_write
+_use_socketfile
 _lowerstr
 _main
 _main_repl
diff --git a/packages/pglite/src/pglite.ts b/packages/pglite/src/pglite.ts
index e267a618a..7128cbc07 100644
--- a/packages/pglite/src/pglite.ts
+++ b/packages/pglite/src/pglite.ts
@@ -299,109 +299,8 @@ export class PGlite
           }
           mod.FS.registerDevice(devId, devOpt)
           mod.FS.mkdev('/dev/blob', devId)
-        },
-        (mod: any) => {
-          console.log('registering SOCKET_FILE.IN device')
-          // Register SOCKET_FILE.IN device
-          const devId = mod.FS.makedev(63, 0)
-          const devOpt = {
-            open: (_stream: any) => {},
-            close: (_stream: any) => {},
-            read: (
-              _stream: any,
-              buffer: Uint8Array,
-              offset: number,
-              length: number,
-              position: number,
-            ) => {
-              const buf = this.#queryInBuffer
-              if (!buf) {
-                throw new Error(`No ${SOCKET_FILE.IN} Buffer provided to read from`)
-              }
-              const contents = new Uint8Array(buf)
-              if (position >= contents.length) return 0
-              const size = Math.min(contents.length - position, length)
-              for (let i = 0; i < size; i++) {
-                buffer[offset + i] = contents[position + i]
-              }
-              return size
-            },
-            write: (
-              _stream: any,
-              _buffer: Uint8Array,
-              _offset: number,
-              _length: number,
-              _position: number,
-            ) => {
-              throw new Error('Not implemented')
-            },
-            llseek: (stream: any, offset: number, whence: number) => {
-              const buf = this.#queryInBuffer
-              if (!buf) {
-                throw new Error(`No ${SOCKET_FILE.IN} Buffer provided to llseek`)
-              }
-              let position = offset
-              if (whence === 1) {
-                position += stream.position
-              } else if (whence === 2) {
-                position = new Uint8Array(buf).length
-              }
-              if (position < 0) {
-                throw new mod.FS.ErrnoError(28)
-              }
-              return position
-            },
-          }
-          console.log('registering SOCKET_FILE.IN device 2')
-          mod.FS.registerDevice(devId, devOpt)
-          console.log('registering SOCKET_FILE.IN device 3')
-          mod.FS.mkdir('/tmp/pglite/base/')
-          mod.FS.mkdev(SOCKET_FILE.IN, devId)
-          console.log('registering SOCKET_FILE.IN device 4')
-          // mod.FS.mkdev(SOCKET_FILE.OLOCK, devId)
-          console.log('registered SOCKET_FILE.IN device')
-        },
-        (mod: any) => {
-          console.log('registering SOCKET_FILE.OUT device')
-          // Register SOCKET_FILE.OUT device
-          const devId = mod.FS.makedev(62, 0)
-          const devOpt = {
-            open: (_stream: any) => {},
-            close: (_stream: any) => {},
-            read: (
-              _stream: any,
-              _buffer: Uint8Array,
-              _offset: number,
-              _length: number,
-              _position: number,
-            ) => {
-              throw new Error('Not implemented')
-            },
-            write: (
-              _stream: any,
-              buffer: Uint8Array,
-              offset: number,
-              length: number,
-              _position: number,
-            ) => {
-              this.#queryOutChunks ??= []
-              this.#queryOutChunks.push(buffer.slice(offset, offset + length))
-              return length
-            },
-            llseek: (_stream: any, _offset: number, _whence: number) => {
-              throw new Error('Not implemented')
-            },
-          }
-          console.log('registering SOCKET_FILE.OUT device 2')
-          mod.FS.registerDevice(devId, devOpt)
-          console.log('registering SOCKET_FILE.OUT device 3')
-          mod.FS.mkdir('/tmp/pglite/base/')
-          mod.FS.mkdev(SOCKET_FILE.OUT, devId)
-          console.log('registering SOCKET_FILE.OUT device 4')
-          // mod.FS.mkdev(SOCKET_FILE.OLOCK, devId)
-          console.log('registered SOCKET_FILE.OUT device')
-        },
-      ],
+        }
+      ]
     }
 
     emscriptenOpts = await this.fs!.emscriptenOpts(emscriptenOpts)
@@ -521,6 +420,9 @@ export class PGlite
     // TODO: only sync here if initdb did init db.
     await this.syncToFs()
 
+    // Setup the socket files for the query protocol IO
+    await this.setupSocketFiles()
+
     this.#ready = true
 
     // Set the search path to public for this connection
@@ -535,6 +437,111 @@ export class PGlite
     }
   }
 
+  async setupSocketFiles() {
+    const mod = this.mod!
+    // Register SOCKET_FILE.IN device
+    console.log('registering SOCKET_FILE.IN device')
+    const inDevId = mod.FS.makedev(63, 0)
+    const inDevOpt = {
+      open: (_stream: any) => {},
+      close: (_stream: any) => {},
+      read: (
+        _stream: any,
+        buffer: Uint8Array,
+        offset: number,
+        length: number,
+        position: number,
+      ) => {
+        console.log('reading SOCKET_FILE.IN device')
+        const buf = this.#queryInBuffer
+        if (!buf) {
+          throw new Error(`No ${SOCKET_FILE.IN} Buffer provided to read from`)
+        }
+        const contents = new Uint8Array(buf)
+        if (position >= contents.length) return 0
+        const size = Math.min(contents.length - position, length)
+        for (let i = 0; i < size; i++) {
+          buffer[offset + i] = contents[position + i]
+        }
+        return size
+      },
+      write: (
+        _stream: any,
+        _buffer: Uint8Array,
+        _offset: number,
+        _length: number,
+        _position: number,
+      ) => {
+        throw new Error('Not implemented')
+      },
+      llseek: (stream: any, offset: number, whence: number) => {
+        console.log('seeking SOCKET_FILE.IN device')
+        const buf = this.#queryInBuffer
+        if (!buf) {
+          throw new Error(`No ${SOCKET_FILE.IN} Buffer provided to llseek`)
+        }
+        let position = offset
+        if (whence === 1) {
+          position += stream.position
+        } else if (whence === 2) {
+          position = new Uint8Array(buf).length
+        }
+        if (position < 0) {
+          throw new mod.FS.ErrnoError(28)
+        }
+        return position
+      },
+    }
+    console.log('registering SOCKET_FILE.IN device 2')
+    mod.FS.registerDevice(inDevId, inDevOpt)
+    console.log('registering SOCKET_FILE.IN device 3')
+    mod.FS.mkdev(SOCKET_FILE.IN, inDevId)
+    console.log('registering SOCKET_FILE.IN device 4')
+    // mod.FS.mkdev(SOCKET_FILE.OLOCK, devId)
+    console.log('registered SOCKET_FILE.IN device')
+
+    // Register SOCKET_FILE.OUT devicex
+    console.log('registering SOCKET_FILE.OUT device')
+    const outDevId = mod.FS.makedev(62, 0)
+    const outDevOpt = {
+      open: (_stream: any) => {},
+      close: (_stream: any) => {},
+      read: (
+        _stream: any,
+        _buffer: Uint8Array,
+        _offset: number,
+        _length: number,
+        _position: number,
+      ) => {
+        throw new Error('Not implemented')
+      },
+      write: (
+        _stream: any,
+        buffer: Uint8Array,
+        offset: number,
+        length: number,
+        _position: number,
+      ) => {
+        console.log('writing SOCKET_FILE.OUT device')
+        this.#queryOutChunks ??= []
+        this.#queryOutChunks.push(buffer.slice(offset, offset + length))
+        return length
+      },
+      llseek: (_stream: any, _offset: number, _whence: number) => {
+        throw new Error('Not implemented')
+      },
+    }
+    console.log('registering SOCKET_FILE.OUT device 2')
+    mod.FS.registerDevice(outDevId, outDevOpt)
+    console.log('registering SOCKET_FILE.OUT device 3')
+    mod.FS.mkdev(SOCKET_FILE.OUT, outDevId)
+    console.log('registering SOCKET_FILE.OUT device 4')
+    // mod.FS.mkdev(SOCKET_FILE.OLOCK, devId)
+    console.log('registered SOCKET_FILE.OUT device')
+
+    mod._use_socketfile()
+  }
+
   /**
    * The Postgres Emscripten Module
    */
diff --git a/packages/pglite/src/postgresMod.ts b/packages/pglite/src/postgresMod.ts
index 5eb7a1eac..7ff99b4f0 100644
--- a/packages/pglite/src/postgresMod.ts
+++ b/packages/pglite/src/postgresMod.ts
@@ -28,6 +28,7 @@ export interface PostgresMod
   _interactive_write: (msgLength: number) => void
   _interactive_one: () => void
   _interactive_read: () => number
+  _use_socketfile: () => void
 }
 
 type PostgresFactory<T extends PostgresMod = PostgresMod> = (
diff --git a/patches/pg_main.c b/patches/pg_main.c
index 4ac51bec0..cb632a099 100644
--- a/patches/pg_main.c
+++ b/patches/pg_main.c
@@ -1182,7 +1182,7 @@ extern void AsyncPostgresSingleUserMain(int single_argc, char *single_argv[], co
 
     #include "../postgresql/src/bin/initdb/initdb.c"
 
-    void use_socketfile(void) {
+    EMSCRIPTEN_KEEPALIVE void use_socketfile(void) {
         is_repl = true;
         is_node = true;
     }