Code: Minor fix