diff --git a/includes/SiteGen/SiteGen.php b/includes/SiteGen/SiteGen.php
index 9c23221..b2484ae 100644
--- a/includes/SiteGen/SiteGen.php
+++ b/includes/SiteGen/SiteGen.php
@@ -50,7 +50,6 @@ class SiteGen {
 	 * Function to check capabilities
 	 */
 	private static function check_capabilities() {
-		return true;
 		$capability = new SiteCapabilities();
 
 		$ai_enabled = $capability->get( 'canAccessAI' );
@@ -212,7 +211,7 @@ private static function generate_pattern_content(
 					'timeout' => 60,
 					'body'    => wp_json_encode(
 						array(
-							'hiivetoken' => 'test-ai-sitegen',
+							'hiivetoken' => HiiveConnection::get_auth_token(),
 							'prompt'     => array(
 								'site_description' => $site_description,
 								'keywords'         => wp_json_encode( $keywords ),
@@ -290,7 +289,7 @@ public static function generate_site_meta( $site_info, $identifier, $skip_cache
 				'timeout' => 60,
 				'body'    => wp_json_encode(
 					array(
-						'hiivetoken' => 'test-ai-sitegen',
+						'hiivetoken' => HiiveConnection::get_auth_token(),
 						'prompt'     => self::get_prompt_from_info( $site_info ),
 						'identifier' => $identifier,
 					)
@@ -381,7 +380,7 @@ public static function get_home_pages( $site_description, $content_style, $targe
 					'timeout' => 60,
 					'body'    => wp_json_encode(
 						array(
-							'hiivetoken' => 'test-ai-sitegen',
+							'hiivetoken' => HiiveConnection::get_auth_token(),
 							'prompt'     => array(
 								'site_description' => $site_description,
 								'keywords'         => wp_json_encode( $keywords ),
@@ -484,7 +483,7 @@ public static function get_content_for_page(
 				'timeout' => 60,
 				'body'    => wp_json_encode(
 					array(
-						'hiivetoken' => 'test-ai-sitegen',
+						'hiivetoken' => HiiveConnection::get_auth_token(),
 						'prompt'     => array(
 							'site_description' => $site_description,
 							'content_style'    => wp_json_encode( $content_style ),