Fix client framework tests
Clarify a TODO in ok.py