Small tweak for map
Merge pull request #218 from gannetson/integration

Small tweak for map