Add a method to prepare for optimizing FlatMapKeys