[FIX] Minor fix