Minor ref improvement