You shouldn't need the `if message` bit here; if `message` is nil then `@expected_message` will be nil, just the same as if you hadn't set it.