Add ref check