Add the missing line in the example code