Minor fix + todo