import される package をカスタマイズする方法はわかりませんが、正しい package を go get してあるのなら、古いキャッシュが影響している可能性があるので、
go clean --modcache
とすると、上手く行くかもしれません。
私の場合は、
pgx.New....
と書いていて、ファイルを保存すると import に
import ( "github.com/jackc/pgx/v4" )
が補完されていたのですが、実際使っているのはv5 なので、毎回書き換えなくてはならず困っていました。
上記コマンドを実行したら無事に v5 が import されるようになりました。