small fix to z_lag (forgot to keep z terms and the other times in the right
proportions)