Removed deprecation of InitialState.FROM_UI and added tests