diff --git a/lora/lora.py b/lora/lora.py
index 5f3a5c01..23a9afdc 100644
--- a/lora/lora.py
+++ b/lora/lora.py
@@ -1,6 +1,5 @@
 # Copyright © 2023-2024 Apple Inc.
 
-import os
 import sys
 import argparse
 import json
@@ -18,10 +17,8 @@ from models import LoRALinear
 
 
 # Disable output buffering to see print statements in real-time
-if sys.version_info >= (3, 7):
-    sys.stdout.reconfigure(line_buffering=True)
-else:
-    sys.stdout = os.fdopen(sys.stdout.fileno(), 'w', 1)
+sys.stdout.reconfigure(line_buffering=True)
+
 
 def build_parser():
     parser = argparse.ArgumentParser(description="LoRA or QLoRA finetuning.")