Review fixes

Signed-off-by: Émile Ré <emile@getprobo.com>

cmd/migrate-document-versions-markdown/main.go

@@ -21,6 +21,7 @@ import (
 	"encoding/json"
 	"flag"
 	"fmt"
+	"net"
 	"net/url"
 	"os"
 	"time"
@@ -130,7 +131,11 @@ func newPgClientFromDSN(dsn string) (*pg.Client, error) {
 	var opts []pg.Option
 
 	if u.Host != "" {
-		opts = append(opts, pg.WithAddr(u.Host))
+		host := u.Host
+		if u.Port() == "" {
+			host = net.JoinHostPort(u.Hostname(), "5432")
+		}
+		opts = append(opts, pg.WithAddr(host))
 	}
 
 	if u.User != nil {