diff --git a/lib.nix b/lib.nix
index cb5b055..9d12207 100644
--- a/lib.nix
+++ b/lib.nix
@@ -87,7 +87,7 @@ in
         )
         // config
         // {
-          unix_socket_directories = "";
+          unix_socket_directories = ".";
           inherit port listen_addresses;
         };
       toStr =
@@ -113,7 +113,6 @@ in
         runtimeInputs = [
           package
           pkgs.coreutils
-          # pkgs.gnused
         ];
         text = ''
           POSTGRES_RUN_INITIAL_SCRIPT="false"
@@ -125,7 +124,6 @@ in
             echo
           fi
           cp ${configFile} "$PGDATA"/postgresql.conf
-          # sed "s|@PGHOST@|''${PGHOST}|g" ${configFile} > "$PGDATA"/postgresql.conf
           if [[ "$POSTGRES_RUN_INITIAL_SCRIPT" == "true" ]]; then
             echo
             echo "PostgreSQL is setting up the initial database"
@@ -137,7 +135,6 @@ in
             echo "Database directory exists. Skipping initialization"
             echo
           fi
-
           unset POSTGRES_RUN_INITIAL_SCRIPT
         '';
       };
@@ -146,9 +143,8 @@ in
         name = "run-postgres";
         text = ''
           set -euo pipefail
-
-          PGDATA=$RUNTIME_PATH/${name}/data
-          PGHOST=$RUNTIME_PATH/${name}/socket
+          PGDATA=$RUNTIME_PATH/${name}
+          PGHOST=$RUNTIME_PATH/${name}
           PGPORT=${builtins.toString port}
           mkdir -p "$PGHOST"
           echo "Starting postgres with PGDATA=''${PGDATA} PGHOST=''${PGHOST} PGPORT=''${PGPORT}"