Added missing piece in code example.