This might work in your testing but you are not supposed to do it ideally. There are few things in IdentityNow which would work if you try to do them but ideally you should not do that in real use case.
Setting Identity attribute from rule is one of those cases. This might break in future. You can contact SP expert services for more detailed explanation on this.