_wait_for_msg() should return message type

[vladak]

While traversing the _wait_for_msg() code, I noticed that there is discrepancy between what the docs say about the return value and reality. This change fixes that.