FIX: Minor fix