Modified for initial state.