update module to use entity reference field