Fixed bug with message receiving