Fix variable assign