Fix documentation after TODO example update