Fixes the code sample in the main README.