diff --git a/spring-boot-samples/spring-boot-sample-data-mongodb/src/test/java/sample/data/mongo/SampleMongoApplicationTests.java b/spring-boot-samples/spring-boot-sample-data-mongodb/src/test/java/sample/data/mongo/SampleMongoApplicationTests.java
index 7dc13a31e6..069434db57 100644
--- a/spring-boot-samples/spring-boot-sample-data-mongodb/src/test/java/sample/data/mongo/SampleMongoApplicationTests.java
+++ b/spring-boot-samples/spring-boot-sample-data-mongodb/src/test/java/sample/data/mongo/SampleMongoApplicationTests.java
@@ -22,6 +22,7 @@ import org.springframework.boot.test.OutputCapture;
 import org.springframework.core.NestedCheckedException;
 
 import com.mongodb.MongoServerSelectionException;
+import com.mongodb.MongoTimeoutException;
 
 import static org.junit.Assert.assertTrue;
 
@@ -55,7 +56,8 @@ public class SampleMongoApplicationTests {
 		NestedCheckedException nested = new NestedCheckedException("failed", ex) {
 		};
 		Throwable root = nested.getRootCause();
-		if (root instanceof MongoServerSelectionException) {
+		if (root instanceof MongoServerSelectionException
+				|| root instanceof MongoTimeoutException) {
 			if (root.getMessage().contains("Unable to connect to any server")) {
 				return true;
 			}