typo + updated TODO