Fix: Fixing Shuffle test case.