Replaced a .should with a expect(...).to