Minor fix of mine