Move out unused import.
Add one more test for the story/replay(proxy=True) usage.