Replace should with expect