diff --git a/lib.nix b/lib.nix
index a6000ae..80830bf 100644
--- a/lib.nix
+++ b/lib.nix
@@ -87,7 +87,7 @@ in
         )
         // config
         // {
-          unix_socket_directories = "../socket";
+          unix_socket_directories = "@PGHOST@";
           inherit port listen_addresses;
         };
       toStr =
@@ -119,6 +119,7 @@ in
           echo "PostgreSQL initdb process completed"
           echo
         fi
+        substituteAllInPlace $PGDATA/postgresql.conf
         cp ${configFile} $PGDATA/postgresql.conf
         if [[ "$POSTGRES_RUN_INITIAL_SCRIPT" == "true" ]]; then
           echo