Are you sure it’s turning off and not just shuffling the power around? AWD systems are not like 4WD systems in that the amount of power sent to any given wheels can vary, depending on the system, some wheelspin may be necessay before the system starts transfering power from the wheels have no grip to the wheels that do.