Update message_test.go