Update sample code to use correct message responses