diff --git a/qemu/tests/balloon_stress.py b/qemu/tests/balloon_stress.py
index c15b2542e16d39a804abc0e1e9075005641dac6a..e89482ba09d9882e667690c3fb4a664e2702fdaa 100644
--- a/qemu/tests/balloon_stress.py
+++ b/qemu/tests/balloon_stress.py
@@ -30,7 +30,6 @@ def run(test, params, env):
     vm.verify_alive()
 
     default_memory = int(params.get("default_memory", params['mem']))
-    unit = vm.monitor.protocol == "qmp" and 1048576 or 1
     timeout = float(params.get("login_timeout", 360))
     session = vm.wait_for_login(timeout=timeout)
     # for media player configuration
@@ -38,9 +37,8 @@ def run(test, params, env):
         session.cmd(params.get("pre_cmd"))
 
     driver_name = params["driver_name"]
-    if params["os_type"] == "windows":
-        utils_test.qemu.setup_win_driver_verifier(driver_name, vm, timeout)
-        balloon_test = BallooningTestWin(test, params, env)
+    utils_test.qemu.setup_win_driver_verifier(driver_name, vm, timeout)
+    balloon_test = BallooningTestWin(test, params, env)
 
     error.context("Play video in guest", logging.info)
     play_video_cmd = params["play_video_cmd"]