Fix: correct variable assignment