Fix up a handful of cases not covered by the merge.

Author: andriyDev

crates/bevy_pbr/src/wireframe.rs

@@ -22,6 +22,7 @@ use bevy_ecs::{
 use bevy_platform::{
     collections::{HashMap, HashSet},
     hash::FixedHasher,
+    sync::Arc,
 };
 use bevy_reflect::{std_traits::ReflectDefault, Reflect};
 use bevy_render::camera::extract_cameras;
@@ -471,7 +472,7 @@ impl RenderAsset for RenderWireframeMaterial {
     type Param = ();
 
     fn prepare_asset(
-        source_asset: Self::SourceAsset,
+        source_asset: Arc<Self::SourceAsset>,
         _asset_id: AssetId<Self::SourceAsset>,
         _param: &mut SystemParamItem<Self::Param>,
     ) -> Result<Self, PrepareAssetError<Self::SourceAsset>> {

crates/bevy_sprite/src/mesh2d/wireframe2d.rs

@@ -22,6 +22,7 @@ use bevy_ecs::{
 use bevy_platform::{
     collections::{HashMap, HashSet},
     hash::FixedHasher,
+    sync::Arc,
 };
 use bevy_reflect::{std_traits::ReflectDefault, Reflect};
 use bevy_render::{
@@ -470,7 +471,7 @@ impl RenderAsset for RenderWireframeMaterial {
     type Param = ();
 
     fn prepare_asset(
-        source_asset: Self::SourceAsset,
+        source_asset: Arc<Self::SourceAsset>,
         _asset_id: AssetId<Self::SourceAsset>,
         _param: &mut SystemParamItem<Self::Param>,
     ) -> Result<Self, PrepareAssetError<Self::SourceAsset>> {

crates/bevy_ui/src/widget/viewport.rs

@@ -171,7 +171,7 @@ pub fn update_viewport_render_target_size(
             height: u32::max(1, size.y as u32),
             ..default()
         };
-        let image = images.get_mut(image_handle).unwrap();
+        let image = images.get_cloned_mut(image_handle).unwrap();
         if image.data.is_some() {
             image.resize(size);
         } else {