Fixes to the example. Thanks @brianchirls!