diff --git a/examples/plot_tasks_MPI.py b/examples/plot_tasks_MPI.py
index c71d96a4e127f348284b7095a612b5783be73e79..b7d1823ad746d6a10b5e67fc9f7315b13be4649f 100755
--- a/examples/plot_tasks_MPI.py
+++ b/examples/plot_tasks_MPI.py
@@ -64,7 +64,7 @@ PLOT_PARAMS = {"axes.labelsize": 10,
 pl.rcParams.update(PLOT_PARAMS)
 
 #  Tasks and subtypes. Indexed as in tasks.h.
-TASKTYPES = ["none", "sort", "self", "pair", "sub", "ghost", "kick1", "kick2",
+TASKTYPES = ["none", "sort", "self", "pair", "sub", "init", "ghost", "drift", "kick",
              "send", "recv", "grav_pp", "grav_mm", "grav_up", "grav_down",
              "psort", "split_cell", "rewait", "count"]
 
@@ -73,9 +73,10 @@ TASKCOLOURS = {"none": "black",
                "self": "greenyellow",
                "pair": "navy",
                "sub": "hotpink",
+               "init": "indigo",
                "ghost": "cyan",
-               "kick1": "maroon",
-               "kick2": "green",
+               "drift": "maroon",
+               "kick": "green",
                "send": "yellow",
                "recv": "magenta",
                "grav_pp": "mediumorchid",