diff --git a/tests/modes/test_modes.py b/tests/modes/test_modes.py
index 2f8bf96..12a4bd6 100644
--- a/tests/modes/test_modes.py
+++ b/tests/modes/test_modes.py
@@ -1,5 +1,4 @@
-from boinc_client import set_cpu_run_mode
-from boinc_client.modes import set_gpu_run_mode, set_network_mode
+from boinc_client.modes import set_cpu_run_mode, set_gpu_run_mode, set_network_mode
 
 
 def test_can_set_cpu_run_mode(mocker, mock_rpc_client):