diff --git a/spring-boot-actuator/src/test/java/org/springframework/boot/actuate/endpoint/InfoEndpointCompatibilityTests.java b/spring-boot-actuator/src/test/java/org/springframework/boot/actuate/endpoint/InfoEndpointCompatibilityTests.java
index 86bec9afd83..85b919366b4 100644
--- a/spring-boot-actuator/src/test/java/org/springframework/boot/actuate/endpoint/InfoEndpointCompatibilityTests.java
+++ b/spring-boot-actuator/src/test/java/org/springframework/boot/actuate/endpoint/InfoEndpointCompatibilityTests.java
@@ -17,6 +17,7 @@
 package org.springframework.boot.actuate.endpoint;
 
 import java.util.Collections;
+import java.util.Map;
 
 import org.junit.Test;
 
@@ -34,7 +35,7 @@ public class InfoEndpointCompatibilityTests {
 
 	@Test
 	public void invoke() throws Exception {
-		Info actual = getEndpointBean().invoke();
+		Map<String, Object> actual = getEndpointBean().invoke();
 		assertThat(actual.get("key1")).isEqualTo("value1");
 		assertThat(actual.get("foo")).isEqualTo("bar");
 	}