Fix bug with colors not matching legend