2 files changed, 19 insertions, 0 deletions
diff --git a/internal/handlers/widget.go b/internal/handlers/widget.go
index 15ef3b2..b8c3aa9 100644
--- a/internal/handlers/widget.go
+++ b/internal/handlers/widget.go
@@ -473,6 +473,10 @@ func (h *Handler) HandleWidgetTaskUpdate(w http.ResponseWriter, r *http.Request)
 		return
 	}
 	if err := h.store.UpdateNativeTask(req.ID, req.Title, req.Description); err != nil {
+		if errors.Is(err, store.ErrNativeTaskNotFound) {
+			http.Error(w, "task not found", http.StatusNotFound)
+			return
+		}
 		http.Error(w, "failed to update task", http.StatusInternalServerError)
 		return
 	}
diff --git a/internal/handlers/widget_test.go b/internal/handlers/widget_test.go
index 35c21e2..95f5264 100644
--- a/internal/handlers/widget_test.go
+++ b/internal/handlers/widget_test.go
@@ -924,6 +924,21 @@ func TestHandleWidgetTaskUpdate_UpdatesTitleAndDescription(t *testing.T) {
 	}
 }
 
+func TestHandleWidgetTaskUpdate_UnknownID_Returns404(t *testing.T) {
+	s, cleanup := setupTestDB(t)
+	defer cleanup()
+	h := &Handler{store: s}
+
+	body := `{"id":"does-not-exist","title":"New title","description":""}`
+	req := httptest.NewRequest("POST", "/api/widget/task/update", strings.NewReader(body))
+	w := httptest.NewRecorder()
+	http.HandlerFunc(h.HandleWidgetTaskUpdate).ServeHTTP(w, req)
+
+	if w.Code != http.StatusNotFound {
+		t.Errorf("expected 404 for an unknown task id, got %d", w.Code)
+	}
+}
+
 func TestHandleWidgetTaskRecurrence_SetsPattern(t *testing.T) {
 	s, cleanup := setupTestDB(t)
 	defer cleanup()