diff --git a/examples/cube.rs b/examples/cube.rs
index 891523f..97ed1f4 100644
--- a/examples/cube.rs
+++ b/examples/cube.rs
@@ -7,7 +7,7 @@ use camera_controllers::*;
 
 fn main() {
     println!("Toggle camera control by pressing C");
-    
+
     let mut capture_cursor = false;
     let (mut window, mut scene, vs, fs) = {
         let settings = WindowSettings::new("colored cube", [512; 2])
@@ -71,8 +71,7 @@ fn main() {
         if capture_cursor {first_person.event(&e)};
 
         if let Some(args) = e.render_args() {
-            let surface_texture = window.surface.get_current_texture().unwrap();
-            scene.state.surface_texture = Some(surface_texture);
+            scene.state.start_render(&window.surface);
 
             scene.clear([0.0, 0.0, 0.0, 1.0]);
 
@@ -83,12 +82,7 @@ fn main() {
 
             scene.draw(cubes, &frame_graph);
 
-            scene.state.end_render_pass();
-            if let Some(surface_texture) = std::mem::replace(
-                &mut scene.state.surface_texture, None
-            ) {
-                surface_texture.present();
-            }
+            scene.state.end_render();
         }
 
         if let Some(button) = e.press_args() {