Fixing some mistake in communication test and adding assertions