diff --git a/nummi/api/views.py b/nummi/api/views.py
index 2d46911..31024eb 100644
--- a/nummi/api/views.py
+++ b/nummi/api/views.py
@@ -44,6 +44,7 @@ class HistoryView(UserMixin, MultipleObjectMixin, View):
             {
                 "data": list(
                     self.get_queryset()
+                    .filter(category__budget=True)
                     .values(month=TruncMonth("date"))
                     .annotate(
                         sum_p=Sum(
diff --git a/nummi/main/forms.py b/nummi/main/forms.py
index 146432a..4ae8339 100644
--- a/nummi/main/forms.py
+++ b/nummi/main/forms.py
@@ -28,6 +28,7 @@ class CategoryForm(NummiForm):
         fields = [
             "name",
             "icon",
+            "budget",
         ]