Fixed broken variable assignment