Fixed little bug with receiving messages