Update factory for new dependency