One more variable assignment fix