modify some test code to add message