Updated contrib module entity reference