Fix reference code sample