From 8a166ac82fa790c4607ed326bddc1e97b3ed5a5c Mon Sep 17 00:00:00 2001
From: acqn <acqn163@outlook.com>
Date: Tue, 11 Feb 2020 18:28:38 +0800
Subject: [PATCH] Fixed register usage tracking interfered by CE_SetArg.

---
 src/cc65/codeent.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/cc65/codeent.c b/src/cc65/codeent.c
index 51f5f1378..7dd90833b 100644
--- a/src/cc65/codeent.c
+++ b/src/cc65/codeent.c
@@ -360,6 +360,10 @@ void CE_SetArg (CodeEntry* E, const char* Arg)
 
     /* Assign the new one */
     E->Arg = GetArgCopy (Arg);
+
+    /* Update the Use and Chg in E */
+    const OPCDesc* D = GetOPCDesc (E->OPC);
+    SetUseChgInfo (E, D);
 }