fix tests
update TODO