diff --git a/coderd/provisionerdserver/provisionerdserver.go b/coderd/provisionerdserver/provisionerdserver.go index b8cf6315a8e3f..0398bbf5f4b3e 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 {