diff --git a/initializr-docs/pom.xml b/initializr-docs/pom.xml
index f172f3cc..24c830c1 100644
--- a/initializr-docs/pom.xml
+++ b/initializr-docs/pom.xml
@@ -74,6 +74,7 @@
 									<attributes>
 										<docinfo>true</docinfo>
 										<spring-boot-version>${spring.boot.version}</spring-boot-version>
+										<snippets>${basedir}/../initializr-web/target/snippets</snippets>
 									</attributes>
 								</configuration>
 							</execution>
diff --git a/initializr-docs/src/test/java/io/spring/initializr/stub/InitializrIntegrationTests.java b/initializr-docs/src/test/java/io/spring/initializr/stub/InitializrIntegrationTests.java
index 496312c2..37c9e89d 100644
--- a/initializr-docs/src/test/java/io/spring/initializr/stub/InitializrIntegrationTests.java
+++ b/initializr-docs/src/test/java/io/spring/initializr/stub/InitializrIntegrationTests.java
@@ -26,8 +26,7 @@ import static org.springframework.boot.test.context.SpringBootTest.*;
 @SpringBootTest(webEnvironment = WebEnvironment.NONE)
 // Need a spring boot app. Doesn't really have one
 @AutoConfigureStubRunner(
-		// TODO: can't hardcode the version there as it must be automatically replaced when releasing
-		ids = "io.spring.initializr:initializr-web:0.3.0.BUILD-SNAPSHOT",
+		ids = "io.spring.initializr:initializr-web",
 		workOffline = true)
 public class InitializrIntegrationTests {
 
diff --git a/initializr-web/src/test/groovy/io/spring/initializr/web/test/MockMvcClientHttpRequestFactoryTestExecutionListener.java b/initializr-web/src/test/groovy/io/spring/initializr/web/test/MockMvcClientHttpRequestFactoryTestExecutionListener.java
index b42788d4..63437a25 100644
--- a/initializr-web/src/test/groovy/io/spring/initializr/web/test/MockMvcClientHttpRequestFactoryTestExecutionListener.java
+++ b/initializr-web/src/test/groovy/io/spring/initializr/web/test/MockMvcClientHttpRequestFactoryTestExecutionListener.java
@@ -39,6 +39,8 @@ public final class MockMvcClientHttpRequestFactoryTestExecutionListener
 					beanFactory.getBean(MockMvc.class));
 			beanFactory.registerSingleton("mockMvcClientHttpRequestFactory",
 					this.factory);
+		} else {
+			factory = beanFactory.getBean("mockMvcClientHttpRequestFactory", MockMvcClientHttpRequestFactory.class);
 		}
 	}