diff --git a/tests/test_shader_live_editor.py b/tests/test_shader_live_editor.py
index 9ad8974..33930ae 100644
--- a/tests/test_shader_live_editor.py
+++ b/tests/test_shader_live_editor.py
@@ -21,5 +21,4 @@ def test_shader_live_editor_renders():
   mock_imscope.style_var.return_value.__enter__.return_value = None
 
   app._render_shader_live_editor()
-  assert mock_imgui.begin.called
-  assert mock_imgui.end.called
+  assert mock_imscope.window.called