fix missing import
Merge pull request #34 from pslegr/ERRAI-470_1

fix missing import
Removed deprecation of InitialState.FROM_UI and added tests