* Fix TODO tests