diff --git a/tests/test_desktop_functions.py b/tests/test_desktop_functions.py
index 2a97137..f012abd 100644
--- a/tests/test_desktop_functions.py
+++ b/tests/test_desktop_functions.py
@@ -95,6 +95,7 @@ def test_create_and_remove_desktop():
 
     time.sleep(1) # Got to wait for the animation before we can return
     current_desktop.go()
+    current_window.set_focus()
 
 
 @pytest.mark.xfail(