diff --git a/sleuth/src/test/java/demo/TraceApplicationTests.java b/sleuth/src/test/java/demo/TraceApplicationTests.java
index 73d44e4..85967db 100644
--- a/sleuth/src/test/java/demo/TraceApplicationTests.java
+++ b/sleuth/src/test/java/demo/TraceApplicationTests.java
@@ -6,8 +6,8 @@ import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.test.SpringApplicationConfiguration;
-import org.springframework.cloud.sleuth.Trace;
-import org.springframework.cloud.sleuth.trace.DefaultTrace;
+import org.springframework.cloud.sleuth.TraceManager;
+import org.springframework.cloud.sleuth.trace.DefaultTraceManager;
 import org.springframework.test.annotation.DirtiesContext;
 import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
 import org.springframework.test.context.web.WebAppConfiguration;
@@ -19,7 +19,7 @@ import org.springframework.test.context.web.WebAppConfiguration;
 public class TraceApplicationTests {
 
 	@Autowired
-	Trace trace;
+	TraceManager traceManager;
 
 	@Test
 	public void contextLoads() {
@@ -27,6 +27,6 @@ public class TraceApplicationTests {
 
 	@Test
 	public void traceIsDefaultTrace() {
-		assertTrue("trace is wrong type", this.trace instanceof DefaultTrace);
+		assertTrue("trace is wrong type", this.traceManager instanceof DefaultTraceManager);
 	}
 }