Extract assignRef function outside the hook

Extract assignRef function outside the hook #151896