diff --git a/persistence-modules/spring-boot-persistence-mongodb/pom.xml b/persistence-modules/spring-boot-persistence-mongodb/pom.xml
index 86b93c7826..585e54bf57 100644
--- a/persistence-modules/spring-boot-persistence-mongodb/pom.xml
+++ b/persistence-modules/spring-boot-persistence-mongodb/pom.xml
@@ -25,6 +25,11 @@
org.springframework.boot
spring-boot-starter-data-mongodb
+
+ de.flapdoodle.embed
+ de.flapdoodle.embed.mongo
+ test
+
org.junit.jupiter
@@ -102,4 +107,4 @@
-
\ No newline at end of file
+
diff --git a/spring-boot/src/test/java/com/baeldung/mongodb/ManualEmbeddedMongoDbIntegrationTest.java b/persistence-modules/spring-boot-persistence-mongodb/src/test/java/com/baeldung/mongodb/ManualEmbeddedMongoDbIntegrationTest.java
similarity index 100%
rename from spring-boot/src/test/java/com/baeldung/mongodb/ManualEmbeddedMongoDbIntegrationTest.java
rename to persistence-modules/spring-boot-persistence-mongodb/src/test/java/com/baeldung/mongodb/ManualEmbeddedMongoDbIntegrationTest.java
diff --git a/spring-boot/src/test/java/com/baeldung/mongodb/MongoDbSpringIntegrationTest.java b/persistence-modules/spring-boot-persistence-mongodb/src/test/java/com/baeldung/mongodb/MongoDbSpringIntegrationTest.java
similarity index 90%
rename from spring-boot/src/test/java/com/baeldung/mongodb/MongoDbSpringIntegrationTest.java
rename to persistence-modules/spring-boot-persistence-mongodb/src/test/java/com/baeldung/mongodb/MongoDbSpringIntegrationTest.java
index 39127f62e9..954bae3684 100644
--- a/spring-boot/src/test/java/com/baeldung/mongodb/MongoDbSpringIntegrationTest.java
+++ b/persistence-modules/spring-boot-persistence-mongodb/src/test/java/com/baeldung/mongodb/MongoDbSpringIntegrationTest.java
@@ -2,7 +2,6 @@ package com.baeldung.mongodb;
import static org.assertj.core.api.Assertions.assertThat;
-import org.baeldung.boot.Application;
import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
@@ -12,10 +11,11 @@ import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit.jupiter.SpringExtension;
+import com.baeldung.SpringBootPersistenceApplication;
import com.mongodb.BasicDBObjectBuilder;
import com.mongodb.DBObject;
-@ContextConfiguration(classes = Application.class)
+@ContextConfiguration(classes = SpringBootPersistenceApplication.class)
@DataMongoTest
@ExtendWith(SpringExtension.class)
public class MongoDbSpringIntegrationTest {
diff --git a/spring-boot/pom.xml b/spring-boot/pom.xml
index 2e463e0189..ed2d8259df 100644
--- a/spring-boot/pom.xml
+++ b/spring-boot/pom.xml
@@ -15,16 +15,7 @@
-
-
- org.springframework.boot
- spring-boot-starter-data-mongodb
-
-
- de.flapdoodle.embed
- de.flapdoodle.embed.mongo
- test
-
+
org.junit.jupiter