diff --git a/includes/SiteGen/SiteGen.php b/includes/SiteGen/SiteGen.php
index 5e9c9f4..3390dfb 100644
--- a/includes/SiteGen/SiteGen.php
+++ b/includes/SiteGen/SiteGen.php
@@ -57,7 +57,9 @@ class SiteGen {
 	 * Function to check capabilities
 	 */
 	private static function check_capabilities() {
-		return true;
+		$capability      = new SiteCapabilities();
+		$sitegen_enabled = $capability->get( 'hasAISiteGen' );
+		return $sitegen_enabled;
 	}
 
 	/**