diff --git a/crud.py b/crud.py
index c268435..89a37ab 100644
--- a/crud.py
+++ b/crud.py
@@ -12,7 +12,8 @@ from .models import (
     NWCBudget,
     NWCKey,
     TrackedSpendNWC,
-    GetNWC
+    GetNWC,
+    NWCNewBudget
 )
 
 db = Database("ext_nwcprovider")
@@ -31,7 +32,7 @@ async def create_nwc(data: CreateNWCKey) -> NWCKey:
     await db.insert("nwcprovider.keys", nwckey_entry)
     if data.budgets:
         for budget in data.budgets:
-            budget_entry = NWCBudget( # fixme
+            budget_entry = NWCNewBudget(  # fixme
                 pubkey=data.pubkey,
                 budget_msats=budget.budget_msats,
                 refresh_window=budget.refresh_window,
@@ -123,15 +124,15 @@ async def get_budgets_nwc(data: GetBudgetsNWC) -> Optional[NWCBudget]:
                 """
                 SELECT SUM(amount_msats) FROM nwcprovider.spent
                 WHERE pubkey = :pubkey AND created_at >=
-                :last_cycle AND created_at < next_cycle
+                :last_cycle AND created_at < :next_cycle
                 """,
                 {
                     "pubkey": data.pubkey,
                     "last_cycle": last_cycle,
                     "next_cycle": next_cycle,
                 },
-            )
-            tot_spent_in_range_msats = tot_spent_in_range_msats[0] or 0
+            )           
+            tot_spent_in_range_msats = next(iter(tot_spent_in_range_msats.values())) or 0
             budget.used_budget_msats = tot_spent_in_range_msats
     return budgets
 
diff --git a/models.py b/models.py
index 5a5556c..6b0baa2 100644
--- a/models.py
+++ b/models.py
@@ -64,10 +64,11 @@ class NWCBudget(BaseModel):
 
 
 class NWCNewBudget(BaseModel):
+    pubkey: Optional[str]
     budget_msats: int
     refresh_window: int
     created_at: int
-
+    
 
 # CRUD models
 class CreateNWCKey(BaseModel):