XD-1389: Added null check when receiving partitioned messages back to handle timeouts