Fix a couple test errors

The StandardError with expects seems to be the reason for failure,
changed it to mock fixed it