diff --git a/tests/Integration/RestAPI/ContentHelper/ContentHelperControllerTest.php b/tests/Integration/RestAPI/ContentHelper/ContentHelperControllerTest.php
index 29f4300f4..8ddcd27e0 100644
--- a/tests/Integration/RestAPI/ContentHelper/ContentHelperControllerTest.php
+++ b/tests/Integration/RestAPI/ContentHelper/ContentHelperControllerTest.php
@@ -15,6 +15,7 @@
 use Parsely\REST_API\Content_Helper\Endpoint_Smart_Linking;
 use Parsely\REST_API\Content_Helper\Endpoint_Excerpt_Generator;
 use Parsely\REST_API\Content_Helper\Endpoint_Title_Suggestions;
+use Parsely\REST_API\Content_Helper\Endpoint_Traffic_Boost;
 use Parsely\Tests\Integration\RestAPI\RestAPIControllerTest;
 use Parsely\Tests\Integration\TestCase;
 
@@ -97,10 +98,11 @@ public function test_init_registers_endpoints(): void {
 		$this->content_helper_controller->init();
 		$endpoints = $this->content_helper_controller->get_endpoints();
 
-		self::assertCount( 3, $endpoints );
+		self::assertCount( 4, $endpoints );
 
 		self::assertInstanceOf( Endpoint_Smart_Linking::class, $endpoints['content-helper/smart-linking'] );
 		self::assertInstanceOf( Endpoint_Excerpt_Generator::class, $endpoints['content-helper/excerpt-generator'] );
 		self::assertInstanceOf( Endpoint_Title_Suggestions::class, $endpoints['content-helper/title-suggestions'] );
+		self::assertInstanceOf( Endpoint_Traffic_Boost::class, $endpoints['content-helper/traffic-boost'] );
 	}
 }