Fix code example in readme.