Fix colors so that identity works with kwargs