[WFLY-3534] - Updating code to reflect changes from transaction subsystem.

This is a patch to a previous PR: https://github.com/wildfly/wildfly/pull/6419.