Set correct colors