fix some stuff with the example