Grouped Draggables With Hidden Rows When Dragging

[hicksoft]

I have rows set up in groups, sort of like a parent-child relationship. When dragging a group, I want to hide all of the children so that only the parents are visible for sorting.

When hiding the children, dnd-kit is unable to accurately track the mouse with the dragged item becoming separated from my cursor. How can I get the dragged item to align with the cursor?

CodeSandbox here https://codesandbox.io/p/sandbox/rx9mxt
You can see the issue when trying to drag Group B or Group C