Correct initialization when the initial state is provided