Fix a missing reference to the map

Hey, here's a small bug I ran into.