Fix end token for Assign nodes
v2.2.4