fixed the code sample to make it fit