diff --git a/spring-boot-project/spring-boot-actuator/src/test/java/org/springframework/boot/actuate/cassandra/CassandraHealthIndicatorTests.java b/spring-boot-project/spring-boot-actuator/src/test/java/org/springframework/boot/actuate/cassandra/CassandraHealthIndicatorTests.java
index 7b9ffa19ea..aacc4e71a7 100644
--- a/spring-boot-project/spring-boot-actuator/src/test/java/org/springframework/boot/actuate/cassandra/CassandraHealthIndicatorTests.java
+++ b/spring-boot-project/spring-boot-actuator/src/test/java/org/springframework/boot/actuate/cassandra/CassandraHealthIndicatorTests.java
@@ -27,6 +27,7 @@ import org.springframework.data.cassandra.core.CassandraOperations;
 import org.springframework.data.cassandra.core.cql.CqlOperations;
 
 import static org.assertj.core.api.Assertions.assertThat;
+import static org.assertj.core.api.Assertions.assertThatIllegalArgumentException;
 import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.BDDMockito.given;
 import static org.mockito.Mockito.mock;
@@ -36,12 +37,12 @@ import static org.mockito.Mockito.mock;
  *
  * @author Oleksii Bondar
  */
-
 public class CassandraHealthIndicatorTests {
 
-	@Test(expected = IllegalArgumentException.class)
-	public void throwsExceptionOnNullCassandraOperations() {
-		new CassandraHealthIndicator(null);
+	@Test
+	public void createWhenCassandraOperationsIsNullShouldThrowException() {
+		assertThatIllegalArgumentException()
+				.isThrownBy(() -> new CassandraHealthIndicator(null));
 	}
 
 	@Test
@@ -54,9 +55,7 @@ public class CassandraHealthIndicatorTests {
 		given(cassandraOperations.getCqlOperations()).willReturn(cqlOperations);
 		given(cqlOperations.queryForResultSet(any(Select.class))).willReturn(resultSet);
 		given(resultSet.isExhausted()).willReturn(true);
-
 		Health health = healthIndicator.health();
-
 		assertThat(health.getStatus()).isEqualTo(Status.UP);
 	}
 
@@ -74,9 +73,7 @@ public class CassandraHealthIndicatorTests {
 		given(resultSet.one()).willReturn(row);
 		String expectedVersion = "1.0.0";
 		given(row.getString(0)).willReturn(expectedVersion);
-
 		Health health = healthIndicator.health();
-
 		assertThat(health.getStatus()).isEqualTo(Status.UP);
 		assertThat(health.getDetails().get("version")).isEqualTo(expectedVersion);
 	}