Updating to new transport message api