diff --git a/classes/Visualizer/Module/AIBuilder.php b/classes/Visualizer/Module/AIBuilder.php index 104c647f..41c24a19 100644 --- a/classes/Visualizer/Module/AIBuilder.php +++ b/classes/Visualizer/Module/AIBuilder.php @@ -369,7 +369,7 @@ public function uploadData(): void { // ── Database query ──────────────────────────────────────────────── case 'db_query': - if ( ! current_user_can( 'manage_options' ) && ! is_super_admin() ) { + if ( ! current_user_can( 'manage_options' ) || ! is_super_admin() || ! Visualizer_Module::is_pro() ) { wp_send_json_error( array( 'message' => __( 'Action not allowed for this user.', 'visualizer' ) ), 403 ); } if ( empty( $_POST['db_query'] ) ) {