Set the active colormap when legend is loaded