Update example

Fixed the import