Update sample due to the modification of unload API