Fix code snippet