There was a problem when pickling TypedImaginaryUnit since it was using the __getnewargs__ of TypedSymbol.