As discussed in https://github.com/microsoft/MLOS/pull/967#issuecomment-2825025256 We need to - [ ] fix the bug - [ ] add tests for this