utils: get subset based on the new citation dictionary format