Fix variable assignment