Fix code example for display