This nessecary here? This should be fine as the discriminator map is at a lower level?