We can get rid of the nested `else` and cut out a line of code here:

```ruby
raise ArgumentError, "Invalid response type :#{type}" if code.nil?
assert_equal code, @response.response_code, message
```