[Transport]Minor Tweaks

Forgot to add this with the previous commit