diff --git a/qemu/tests/cfg/win_virtio_driver_update_test.cfg b/qemu/tests/cfg/win_virtio_driver_update_test.cfg
index 0a4070ccea4135bf1d6b8c334356f13d9e48e5c1..f31725dfdd03de19a8ed8bc8c9711fa3e9002044 100644
--- a/qemu/tests/cfg/win_virtio_driver_update_test.cfg
+++ b/qemu/tests/cfg/win_virtio_driver_update_test.cfg
@@ -58,6 +58,9 @@
             test_tags = "evict enlarge"
             balloon_type_evict = evict
             balloon_type_enlarge = enlarge
+            bg_stress_run_flag = balloon_test
+            set_bg_stress_flag = yes
+            wait_bg_time = 720
         - with_netkvm:
             driver_name = netkvm
             nics += " nic2"
diff --git a/qemu/tests/driver_in_use.py b/qemu/tests/driver_in_use.py
index 3b25908efb46f5ea4054e01b665369376f46e4e6..145a62ceb7745c41d5dbb8960df7699f19e62d18 100644
--- a/qemu/tests/driver_in_use.py
+++ b/qemu/tests/driver_in_use.py
@@ -19,7 +19,7 @@ def check_bg_running(vm, params):
              else return False
     """
     session = vm.wait_for_login()
-    target_process = params["target_process"]
+    target_process = params.get("target_process", "")
     if params['os_type'] == 'linux':
         output = session.cmd_output_safe('pgrep -l %s' % target_process)
     else:
@@ -58,7 +58,6 @@ def run(test, params, env):
                               logging.info)
         stress_thread = None
         wait_time = float(params.get("wait_bg_time", 60))
-        target_process = params.get("target_process", "")
         bg_stress_run_flag = params.get("bg_stress_run_flag")
         # Need to set bg_stress_run_flag in some cases to make sure all
         # necessary steps are active