[2.2] TODO fixes

This PR resolves all the remaining TODO 2.2 comments.
@rickardoberg Why was this assertion removed?