Fix choosing colors in 3D

Fixes #297.