maybe stageIdToPool, now that you're indexing by the ID? same below?