Fixes in the code snippet