Use is_amp_endpoint() instead of looking at query var directly

xwp · Dec 7, 2017 · cdc7080 · cdc7080
1 parent 60622f1
commit cdc7080
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/php/class-wp-customize-posts-preview.php b/php/class-wp-customize-posts-preview.php
@@ -166,9 +166,9 @@ public function sanitize_amp_rendered_content_partial( $rendered, $partial ) {
 			&&
 			function_exists( 'amp_load_classes' )
 			&&
-			defined( 'AMP_QUERY_VAR' )
+			function_exists( 'is_amp_endpoint' )
 			&&
-			false !== get_query_var( AMP_QUERY_VAR, false )
+			is_amp_endpoint()
 		);
 		if ( $should_get_amp_content ) {
 			$post = get_post( $partial->post_id );