Replaced sinon.assert call with expect()s