Improved unit tests for the communication package