Minor fix(es)