change getModelsForAffect() method