Forgot to update the top part of the example