diff --git a/src/wp-includes/media.php b/src/wp-includes/media.php
index 8b8b862e5d..3d8507ff90 100644
--- a/src/wp-includes/media.php
+++ b/src/wp-includes/media.php
@@ -4029,9 +4029,11 @@ function wp_prepare_attachment_for_js( $attachment ) {
 		$response['compat'] = get_compat_media_markup( $attachment->ID, array( 'in_modal' => true ) );
 	}
 
-	$media_states = get_media_states( $attachment );
-	if ( ! empty( $media_states ) ) {
-		$response['mediaStates'] = implode( ', ', $media_states );
+	if ( function_exists( 'get_media_states' ) ) {
+		$media_states = get_media_states( $attachment );
+		if ( ! empty( $media_states ) ) {
+			$response['mediaStates'] = implode( ', ', $media_states );
+		}
 	}
 
 	/**