Reverted change to transformer