BUG: bring back which_lag