Minor changes on test to use proper message