diff --git a/pkg/probod/probod.go b/pkg/probod/probod.go index 53e714da3..592590f70 100644 --- a/pkg/probod/probod.go +++ b/pkg/probod/probod.go @@ -277,6 +277,8 @@ func (impl *Implm) Run( wg.Wait() + pgClient.Close() + return context.Cause(ctx) }