tested with large storage. put multiple stacks of different items into different storage units, pull out one stack at the control block and the rest disappears.

Looks like this only affects shift clicking out of the block, just taking the stack out works correctly for me, but I can certainly recreate the issue in dev and will fix it soon.