XPS 9560 Usb C Sleep issues

So I have got Big Sur working almost perfectly with Hackintosh. However, when it goes to sleep, the usb c port doesn't seem to be able to read the hard drive anymore.

It says usb ejected unsafely, which appears to be a relatively common issue, but the fact it doesn't come back is more frustrating.

Anyone had similar problems / a good fix for it?

Thanks

Author: @JackED42