Update sample code to run.