What's the logic for zeroing the `uaux`? The `j` term has `.subs(uaz)` but not the `k` one.