I think thats correct behaviour. When performing TX over USB (via R-PI) you wont get a waterfall until you unkey. Reason I think is that USB has a small amount of bandwidth, and having waterfall data going over USB while TX audio is also going over USB, makes for contention on the USB bus for resources.