moving mock query and changing entity ref