[2.2] TODO fixes

This PR resolves all the remaining TODO 2.2 comments.