Fixed unit display counts so that they are the correct colors