diff --git a/tasks/makespan/run.py b/tasks/makespan/run.py
index 0905fc4..d6c2997 100644
--- a/tasks/makespan/run.py
+++ b/tasks/makespan/run.py
@@ -63,7 +63,7 @@ def granny(
     num_vms=32,
     num_cpus_per_vm=8,
     num_tasks=100,
-    migrate=False
+    migrate=False,
 ):
     """
     Run: `inv makespan.run.granny --workload [mpi,mpi-migrate,mpi-no-migrate]
diff --git a/tasks/makespan/trace.py b/tasks/makespan/trace.py
index c2e5f01..cfa7782 100644
--- a/tasks/makespan/trace.py
+++ b/tasks/makespan/trace.py
@@ -78,9 +78,7 @@ def generate(ctx, workload, num_tasks, num_cores_per_vm, lmbd="0.1"):
     lmbd = float(lmbd)
 
     # Work out the possible number of cores per VM
-    possible_mpi_sizes = arange(
-        2, int(num_cores_per_vm * 2)
-    )
+    possible_mpi_sizes = arange(2, int(num_cores_per_vm * 2))
     possible_omp_sizes = arange(1, num_cores_per_vm)
 
     # The lambda parameter regulates how frequently new tasks arrive. If we