[#3] Fix for message_test.py