Fix issue w/ assigning colors fixes #241