Hmm… I have just tried to replicate the issue you are having, but everything works as expected in my test cases.
Please see the attached gif for the example tree I tested. Can you please explain to me a bit more what the problem is or a way to reproduce that problem more specifically?