Oops! Minor fix