Match new functionality

Updated example to match new functionality