Fix the code samples