diff --git a/wp-modules/pattern-data-handlers/pattern-data-handlers.php b/wp-modules/pattern-data-handlers/pattern-data-handlers.php
index fd35712f7..880ac29bd 100644
--- a/wp-modules/pattern-data-handlers/pattern-data-handlers.php
+++ b/wp-modules/pattern-data-handlers/pattern-data-handlers.php
@@ -372,7 +372,7 @@ function tree_shake_theme_images() {
 	$backedup_images_dir = $wp_filesystem->wp_content_dir() . 'temp-images/';
 	$images_dir          = $theme_dir . '/patterns/images/';
 
-	$wp_theme_url = get_template_directory_uri();
+	$wp_theme_url = get_stylesheet_directory_uri();
 	$images_url   = $wp_theme_url . '/patterns/images/';
 
 	if ( ! $wp_filesystem->exists( $backedup_images_dir ) ) {
@@ -440,7 +440,7 @@ function move_block_images_to_theme( $pattern_html ) {
 	$assets_dir   = $wp_theme_dir . '/assets/';
 	$images_dir   = $wp_theme_dir . '/patterns/images/';
 
-	$wp_theme_url = get_template_directory_uri();
+	$wp_theme_url = get_stylesheet_directory_uri();
 	$images_url   = $wp_theme_url . '/patterns/images/';
 
 	if ( ! $wp_filesystem->exists( $assets_dir ) ) {
@@ -492,7 +492,7 @@ function move_block_images_to_theme( $pattern_html ) {
 		);
 
 		// Replace the URL with the one we just added to the theme.
-		$pattern_html = str_replace( $url_found, "<?php echo esc_url( get_template_directory_uri() ); ?>/patterns/images/$filename", $pattern_html );
+		$pattern_html = str_replace( $url_found, "<?php echo esc_url( get_stylesheet_directory_uri() ); ?>/patterns/images/$filename", $pattern_html );
 	}
 
 	return $pattern_html;