Minor fix on database