Fix import in example.