Fixing coloring bug