Fixed example

Example now adheres to the correct api