Update the benchmarks and unit tests to use the new StateConfig struct.