Fix buggy colors

This seems to resolve issue #2