Fix the code sample in the README