Minor fix - Variable rename