Fix spec; Add TODO