I wonder if we can push this context logic into `getGraphStore`, but it's a property of Dataset..