From bc0b6d80b2f715f50f2ae83c7df4da7ab14774b8 Mon Sep 17 00:00:00 2001
From: Spencer Gibb <spencer@gibb.us>
Date: Thu, 12 Jan 2017 15:20:58 -0700
Subject: [PATCH] polish

---
 .../ZookeeperAutoServiceRegistrationAutoConfiguration.java    | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/spring-cloud-zookeeper-discovery/src/main/java/org/springframework/cloud/zookeeper/serviceregistry/ZookeeperAutoServiceRegistrationAutoConfiguration.java b/spring-cloud-zookeeper-discovery/src/main/java/org/springframework/cloud/zookeeper/serviceregistry/ZookeeperAutoServiceRegistrationAutoConfiguration.java
index 669cc9ad..c5ad0a99 100644
--- a/spring-cloud-zookeeper-discovery/src/main/java/org/springframework/cloud/zookeeper/serviceregistry/ZookeeperAutoServiceRegistrationAutoConfiguration.java
+++ b/spring-cloud-zookeeper-discovery/src/main/java/org/springframework/cloud/zookeeper/serviceregistry/ZookeeperAutoServiceRegistrationAutoConfiguration.java
@@ -53,8 +53,8 @@ public class ZookeeperAutoServiceRegistrationAutoConfiguration {
 	}
 
 	@Bean
-	@ConditionalOnMissingBean
-	public ZookeeperRegistration zookeeperRegistration(ApplicationContext context, ZookeeperDiscoveryProperties properties) {
+	@ConditionalOnMissingBean(ZookeeperRegistration.class)
+	public ServiceInstanceRegistration zookeeperRegistration(ApplicationContext context, ZookeeperDiscoveryProperties properties) {
 		String host = properties.getInstanceHost();
 		if (!StringUtils.hasText(host)) {
 			throw new IllegalStateException("instanceHost must not be empty");