please add a method which gets instantiated models, so we don't need this duplicated logic.