Minor fix once again