Fixed small bug in map