diff --git a/python/swiftestio/swiftestio.py b/python/swiftestio/swiftestio.py
index 9da200cb4..69c88b711 100644
--- a/python/swiftestio/swiftestio.py
+++ b/python/swiftestio/swiftestio.py
@@ -710,11 +710,10 @@ def swiftest_xr2_infile(ds, config, framenum=-1):
             print(pli['Radius'].values, file=plfile)
          print(pli['px'].values, pli['py'].values, pli['pz'].values, file=plfile)
          print(pli['vx'].values, pli['vy'].values, pli['vz'].values, file=plfile)
-      if config['ROTATION'] == 'YES':
-         print(pli['Ip_x'].values, pli['Ip_y'].values, pli['Ip_z'].values, file=plfile)
-         print(pli['rot_x'].values, pli['rot_y'].values, pli['rot_z'].values, file=plfile)
-
-         plfile.close()
+         if config['ROTATION'] == 'YES':
+            print(pli['Ip_x'].values, pli['Ip_y'].values, pli['Ip_z'].values, file=plfile)
+            print(pli['rot_x'].values, pli['rot_y'].values, pli['rot_z'].values, file=plfile)
+      plfile.close()
 
       # TP file
       tpfile = open(config['TP_IN'], 'w')