### What problem does this PR solve? Remove unstable assertion introduced in PR #5924 ### Type of change - [x] update test cases