diff --git a/swf_filter.py b/swf_filter.py
index 790b606609bbe6dd097b9eef37757e3b5ca5a1fb..fbd043dbb4a249f09023f37e064011e30bbe1df4 100755
--- a/swf_filter.py
+++ b/swf_filter.py
@@ -92,7 +92,7 @@ def main():
     """
     Program entry point.
 
-    Parses the input arguments then calls generate_flat_platform.
+    Parses the input arguments then calls filter_workload.
     """
     parser = argparse.ArgumentParser(
         description="Filter jobs from the SWF input according to some criteria "
@@ -105,12 +105,15 @@ def main():
                         help='The optional output SWF file')
     parser.add_argument('-sp', '--partitions_to_select',
                         type=int, nargs='+', default=None,
-                        help='List of partitions to only consider in the input trace. The jobs running in the other partitions will be discarded.')
+                        help="List of partitions to only consider in the input "
+                        "trace. The jobs running in the other partitions will "
+                        "be discarded.")
 
     parser.add_argument('--keep_only',
                         type=str,
                         default=None,
-                        help='If set, this parameter is evaluated to choose which jobs should be kept')
+                        help="If set, this parameter is evaluated to choose "
+                        "which jobs should be kept")
 
     parser.add_argument("-q", "--quiet", action="store_true")