-LRN: fix integration tests

1 files changed, 6 insertions, 6 deletions

diff --git a/src/integration-tests/test_integration_disconnect.py.in b/src/integration-tests/test_integration_disconnect.py.in
index afeb5a968..2a863f8b2 100755
--- a/src/integration-tests/test_integration_disconnect.py.in
+++ b/src/integration-tests/test_integration_disconnect.py.in
@@ -45,14 +45,14 @@ testname = "test_integration_disconnect"
 verbose = True
 check_timeout = 180
 
+if os.name == "nt":
+  tmp = os.getenv ("TEMP")
+else:
+  tmp = "/tmp"
 
 def cleanup ():
-        if os.name == "nt":
-            shutil.rmtree (os.path.join (os.getenv ("TEMP"), "c_bootstrap_server"), True)
-            shutil.rmtree (os.path.join (os.getenv ("TEMP"), "c_no_nat_client"), True)
-        else:
-            shutil.rmtree ("/tmp/c_bootstrap_server/", True)
-            shutil.rmtree ("/tmp/c_no_nat_client/", True)    
+    shutil.rmtree (os.path.join (tmp, "c_bootstrap_server"), True)
+    shutil.rmtree (os.path.join (tmp, "c_no_nat_client"), True)
 
 
 def success_disconnect_cont (check):