Fix variable confusion