diff --git a/coderd/agentapi/lifecycle.go b/coderd/agentapi/lifecycle.go
index 9c34b2c548..de9d4bd105 100644
--- a/coderd/agentapi/lifecycle.go
+++ b/coderd/agentapi/lifecycle.go
@@ -112,7 +112,7 @@ func (a *LifecycleAPI) UpdateLifecycle(ctx context.Context, req *agentproto.Upda
 		ReadyAt:        readyAt,
 	})
 	if err != nil {
-		if !xerrors.Is(err, context.Canceled) {
+		if !database.IsQueryCanceledError(err) {
 			// not an error if we are canceled
 			logger.Error(ctx, "failed to update lifecycle state", slog.Error(err))
 		}