Resolve false positive deepspeed unit test failure in edge case