TreeGrid ComponentRenderer not focusable ?

I have succeeded in properly displaying TextFields in the editable part of my PropertyGrid.
But when some changes result in data being reloaded, the focus is lost.
I tried keeping a reference to TextFields in the grid and called focus() to try and refocus the field but it doesn't do anything.

Am I missing something ?

Thanks!
 

I've reported it as a bug on github because I think this should work

For reference, the issue is https://github.com/vaadin/framework/issues/8995 (still open).