correcting some stuff for the example