From d03c3cd6278bbd66e1b59c9013bca3a6a8bd3c02 Mon Sep 17 00:00:00 2001
From: Spencer Gibb <spencer@gibb.us>
Date: Thu, 7 Feb 2019 15:36:41 -0500
Subject: [PATCH] Fixes class type tests

---
 .../consul/config/ConsulPropertySourceLocatorFilesTests.java | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/spring-cloud-consul-config/src/test/java/org/springframework/cloud/consul/config/ConsulPropertySourceLocatorFilesTests.java b/spring-cloud-consul-config/src/test/java/org/springframework/cloud/consul/config/ConsulPropertySourceLocatorFilesTests.java
index e433d183..04c8c5d6 100644
--- a/spring-cloud-consul-config/src/test/java/org/springframework/cloud/consul/config/ConsulPropertySourceLocatorFilesTests.java
+++ b/spring-cloud-consul-config/src/test/java/org/springframework/cloud/consul/config/ConsulPropertySourceLocatorFilesTests.java
@@ -112,8 +112,7 @@ public class ConsulPropertySourceLocatorFilesTests {
 		PropertySource<?> bootstrapProperties = propertySources
 				.get("bootstrapProperties");
 		assertThat(bootstrapProperties).as("bootstrapProperties was null").isNotNull();
-		assertThat(bootstrapProperties.getClass())
-				.as("bootstrapProperties was wrong type")
+		assertThat(bootstrapProperties).as("bootstrapProperties was wrong type")
 				.isInstanceOf(CompositePropertySource.class);
 
 		Collection<PropertySource<?>> consulSources = ((CompositePropertySource) bootstrapProperties)
@@ -121,7 +120,7 @@ public class ConsulPropertySourceLocatorFilesTests {
 		assertThat(consulSources).as("consulSources was wrong size").hasSize(1);
 
 		PropertySource<?> consulSource = consulSources.iterator().next();
-		assertThat(consulSource.getClass()).as("consulSource was wrong type")
+		assertThat(consulSource).as("consulSource was wrong type")
 				.isInstanceOf(CompositePropertySource.class);
 		Collection<PropertySource<?>> fileSources = ((CompositePropertySource) consulSource)
 				.getPropertySources();