diff --git a/product_management/core.py b/product_management/core.py
index 52a371a..1e01418 100644
--- a/product_management/core.py
+++ b/product_management/core.py
@@ -1627,7 +1627,8 @@ WHERE resource_ref_id = ${llm_id}$ AND product_type = 'llm_model' LIMIT 1"""
                 luid = getattr(llmusage, 'id', None)
                 if luid:
                     from llmage.balance import finalize_balance
-                    await finalize_balance(ServerEnv(), luid, customer_amount)
+                    await finalize_balance(ServerEnv(), luid, customer_amount,
+                                           llmid=getattr(llmusage, 'llmid', None))
             except Exception as e:
                 exception(f'finalize_balance failed (non-critical): {e}')