Update to follow sample code.