Minor fix of registration