Wrong InitialState in Commonroad Search Tutorial

Hey, I tried to execute cell 3.4. in tutorial_commonroad_search.ipynb. When doing this, I am facing an error:

AssertionError: <Obstacle/initial_state>: argument initial_state of wrong type. 
Expected types: <class 'commonroad.scenario.state.InitialState'>.
Got type: <class 'commonroad.scenario.state.KSState'>.

I haven’t changed anything in the code base yet, I just wanted to run the tutorial. Do you have any suggestions on how to fix this error?