diff --git a/python/ray/tune/BUILD b/python/ray/tune/BUILD
index 1a554ccb2..4327b6777 100644
--- a/python/ray/tune/BUILD
+++ b/python/ray/tune/BUILD
@@ -7,7 +7,7 @@ py_test(
     name = "test_actor_reuse",
     size = "medium",
     srcs = ["tests/test_actor_reuse.py"],
-    tags = ["jenkins_only"],
+    tags = ["exclusive"],
     deps = [":tune_lib"],
 )
 
diff --git a/python/ray/tune/trainable.py b/python/ray/tune/trainable.py
index cf6d7c0b3..1cf326ff2 100644
--- a/python/ray/tune/trainable.py
+++ b/python/ray/tune/trainable.py
@@ -398,9 +398,9 @@ class Trainable:
 
         self.log_result(result)
 
-        if self._stdout_stream:
+        if self._stdout_context:
             self._stdout_stream.flush()
-        if self._stderr_stream:
+        if self._stderr_context:
             self._stderr_stream.flush()
 
         return result