diff --git a/src/fix_get_types.py b/src/fix_get_types.py
index f5df60522..208b9dfcd 100644
--- a/src/fix_get_types.py
+++ b/src/fix_get_types.py
@@ -1,15 +1,15 @@
+#!/usr/bin/env python3
+
 import re
 import argparse
 
-if __name__=='__main__':
-    parser = argparse.ArgumentParser()
-    parser.add_argument('input')
-    parser.add_argument('output')
-    args = parser.parse_args()
+parser = argparse.ArgumentParser ()
+parser.add_argument ('input')
+parser.add_argument ('output')
+args = parser.parse_args ()
 
-    with open(args.input, 'r') as inp:
-        with open(args.output, 'w') as out:
-            for l in inp.readlines():
-                l = re.sub('_t_get_type', '_get_type', l)
-                l = re.sub('_T \(', ' (', l)
-                out.write(l)
+with open (args.input, 'r') as inp, open (args.output, 'w') as out:
+    for l in inp.readlines ():
+        l = re.sub ('_t_get_type', '_get_type', l)
+        l = re.sub ('_T \(', ' (', l)
+        out.write (l)