Small fixup

Looks nicer I guess