fix typo + TODO