use colormap from correct axes environment

Fixes bug #320.