diff --git a/coderd/provisionerdserver/provisionerdserver.go b/coderd/provisionerdserver/provisionerdserver.go index 01a377503d..8cfc4a176f 100644 --- a/coderd/provisionerdserver/provisionerdserver.go +++ b/coderd/provisionerdserver/provisionerdserver.go @@ -368,7 +368,7 @@ func (s *server) AcquireJobWithCancel(stream proto.DRPCProvisionerDaemon_Acquire je = <-jec case je = <-jec: } - if xerrors.Is(je.err, context.Canceled) { + if database.IsQueryCanceledError(je.err) { s.Logger.Debug(streamCtx, "successful cancel") err := stream.Send(&proto.AcquiredJob{}) if err != nil {