2 files changed, 6 insertions, 10 deletions

diff --git a/src/lib.rs b/src/lib.rs
index 28eecd8..0f69fd3 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -33,7 +33,7 @@ impl Plugin for DitherPostProcessPlugin {
 
         app.add_plugins((ExtractComponentPlugin::<DitherPostProcessSettings>::default(),));
 
-        let Ok(render_app) = app.get_sub_app_mut(RenderApp) else {
+        let Some(render_app) = app.get_sub_app_mut(RenderApp) else {
             return;
         };
 
@@ -53,7 +53,7 @@ impl Plugin for DitherPostProcessPlugin {
     }
 
     fn finish(&self, app: &mut App) {
-        let Ok(render_app) = app.get_sub_app_mut(RenderApp) else {
+        let Some(render_app) = app.get_sub_app_mut(RenderApp) else {
             return;
         };
 
diff --git a/src/nodes.rs b/src/nodes.rs
index 377718a..1cf3972 100644
--- a/src/nodes.rs
+++ b/src/nodes.rs
@@ -2,14 +2,10 @@ use bevy::{
     ecs::query::QueryItem,
     prelude::*,
     render::{
-        render_asset::RenderAssets,
-        render_graph::{NodeRunError, RenderGraphContext, RenderLabel, ViewNode},
-        render_resource::{
+        render_asset::RenderAssets, render_graph::{NodeRunError, RenderGraphContext, RenderLabel, ViewNode}, render_resource::{
             BindGroupEntries, Operations, PipelineCache, RenderPassColorAttachment,
             RenderPassDescriptor,
-        },
-        renderer::RenderContext,
-        view::ViewTarget,
+        }, renderer::RenderContext, texture::GpuImage, view::ViewTarget
     },
 };
 
@@ -46,8 +42,8 @@ impl ViewNode for DitherRenderNode {
         let post_process = view_target.post_process_write();
 
         let Some(threshold_map) = world
-            .resource::<RenderAssets<Image>>()
-            .get(dither_post_process_settings.handle())
+            .resource::<RenderAssets<GpuImage>>()
+            .get(dither_post_process_settings.handle().id())
         else {
             warn!("Failed to get threshold map, skipping...");
             return Ok(());