diff --git a/bootstrap.php b/bootstrap.php
index f7739a0..5a95401 100644
--- a/bootstrap.php
+++ b/bootstrap.php
@@ -11,10 +11,6 @@
 		'plugins_loaded',
 		function () {
 			// Set Global Constants
-			if ( ! defined( 'NFD_MODULE_AI_VERSION' ) ) {
-				define( 'NFD_MODULE_AI_VERSION', '0.0.1' );
-			}
-
 			if ( ! defined( 'NFD_MODULE_AI_DIR' ) ) {
 				define( 'NFD_MODULE_AI_DIR', __DIR__ );
 			}
diff --git a/includes/Utils/AISearchUtil.php b/includes/Utils/AISearchUtil.php
index 9ed8bd4..fc744aa 100644
--- a/includes/Utils/AISearchUtil.php
+++ b/includes/Utils/AISearchUtil.php
@@ -69,7 +69,7 @@ public static function get_search_results(
 
 		try {
 			return array(
-				'result'  => $parsed_response['payload']['choices'][0]['text'],
+				'result'  => $parsed_response['payload']['choices'],
 				'post_id' => $parsed_response['payload']['postId'],
 			);
 		} catch ( \Exception $exception ) {