diff --git a/llmage/accounting.py b/llmage/accounting.py
index 986bc19..d577097 100644
--- a/llmage/accounting.py
+++ b/llmage/accounting.py
@@ -27,8 +27,7 @@ async def llm_charging(ppid, llmusage):
 		amount += p.amount
 		if p.cost:
 			cost += p.cost
-	discount = await env.sor_get_customer_discount(sor, 
-				llmusage.ownerid, 
+	discount = await env.get_customer_discount(llmusage.ownerid, 
 				llmusage.userorgid)
 	return DictObject(**{
 		'original_amount': amount,