diff --git a/postgresql.go b/postgresql.go
index e6c60bd..d6992f5 100644
--- a/postgresql.go
+++ b/postgresql.go
@@ -5,7 +5,7 @@ import (
 	"fmt"
 	"time"
 
-	_ "github.com/jackc/pgx/v4/stdlib"
+	"github.com/jackc/pgx/v4/stdlib"
 	"github.com/volatiletech/sqlboiler/v4/boil"
 	"go.elastic.co/apm/module/apmsql/v2"
 	_ "go.elastic.co/apm/module/apmsql/v2/pgxv4"
@@ -32,6 +32,7 @@ func Connect(cfg Config, server string) (db *sql.DB, err error) {
 
 	// connect
 	if cfg.UseElasticAPM {
+		apmsql.Register("pgx", &stdlib.Driver{})
 		db, err = apmsql.Open("pgx", uri)
 	} else {
 		db, err = sql.Open("pgx", uri)