Fix code sample issue