Handle the identity class being passed into the generator