Fix test to use message