probably better test

[XinyuLiu5566]

When I run pytest --flake-finder, this test gave me an error about the first assertion because in the last run this is already in the dictionary, so I think it might be better to delete the key at the end of the test so if people who want to find flaky test run this tool again in the future they won't get the error. That's the reason I open this PR.