diff --git a/pom.xml b/pom.xml
index 4b2f41a..258b8b4 100644
--- a/pom.xml
+++ b/pom.xml
@@ -40,6 +40,11 @@
 			<artifactId>spring-boot-starter-test</artifactId>
 			<scope>test</scope>
 		</dependency>
+		<dependency>
+			<groupId>org.springframework.shell</groupId>
+			<artifactId>spring-shell-starter-test</artifactId>
+			<scope>test</scope>
+		</dependency>
 		<dependency>
 			<groupId>io.projectreactor</groupId>
 			<artifactId>reactor-test</artifactId>
diff --git a/src/test/java/dev/danvega/dadjoke/DadjokeApplicationTests.java b/src/test/java/dev/danvega/dadjoke/DadjokeApplicationTests.java
index d9a1914..34b1fa4 100644
--- a/src/test/java/dev/danvega/dadjoke/DadjokeApplicationTests.java
+++ b/src/test/java/dev/danvega/dadjoke/DadjokeApplicationTests.java
@@ -1,9 +1,9 @@
 package dev.danvega.dadjoke;
 
 import org.junit.jupiter.api.Test;
-import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.shell.test.autoconfigure.ShellTest;
 
-@SpringBootTest
+@ShellTest
 class DadjokeApplicationTests {
 
 	@Test