Same thing here. Have this if/else instruction return the Map.