updating unit tests for message