Fix receive message issue