Updated TimeStepConfusion with new call signature