Fix graph rendering when current state is retried