Fix unit tests to checking createStage() action