Add TODO.
Fix invalid object returned.