Replacing should with expect.