Bug fix for steps function in unit test which should return state instead of transaction

Change-Id: Ie4a5e07bf30a3aa359630ec31b8d20e040c37dbf