templates: use short_ref instead of manually checking ref type

There's a function that does this already.