diff --git a/tfhe/src/high_level_api/integers/signed/base.rs b/tfhe/src/high_level_api/integers/signed/base.rs
index d279c7c66d..4aaa908a4a 100644
--- a/tfhe/src/high_level_api/integers/signed/base.rs
+++ b/tfhe/src/high_level_api/integers/signed/base.rs
@@ -297,7 +297,7 @@ where
                     crate::FheUint32Id::num_blocks(cuda_key.key.message_modulus),
                     streams,
                 );
-                crate::FheUint32::new(result)
+                crate::FheUint32::new(result, cuda_key.tag.clone())
             }),
         })
     }
@@ -341,7 +341,7 @@ where
                     crate::FheUint32Id::num_blocks(cuda_key.key.message_modulus),
                     streams,
                 );
-                crate::FheUint32::new(result)
+                crate::FheUint32::new(result, cuda_key.tag.clone())
             }),
         })
     }
@@ -385,7 +385,7 @@ where
                     crate::FheUint32Id::num_blocks(cuda_key.key.message_modulus),
                     streams,
                 );
-                crate::FheUint32::new(result)
+                crate::FheUint32::new(result, cuda_key.tag.clone())
             }),
         })
     }
@@ -429,7 +429,7 @@ where
                     crate::FheUint32Id::num_blocks(cuda_key.key.message_modulus),
                     streams,
                 );
-                crate::FheUint32::new(result)
+                crate::FheUint32::new(result, cuda_key.tag.clone())
             }),
         })
     }
@@ -473,7 +473,7 @@ where
                     crate::FheUint32Id::num_blocks(cuda_key.key.message_modulus),
                     streams,
                 );
-                crate::FheUint32::new(result)
+                crate::FheUint32::new(result, cuda_key.tag.clone())
             }),
         })
     }
@@ -526,7 +526,10 @@ where
                     crate::FheUint32Id::num_blocks(cuda_key.key.message_modulus),
                     streams,
                 );
-                (crate::FheUint32::new(result), FheBool::new(is_ok))
+                (
+                    crate::FheUint32::new(result, cuda_key.tag.clone()),
+                    FheBool::new(is_ok, cuda_key.tag.clone()),
+                )
             }),
         })
     }
diff --git a/tfhe/src/high_level_api/integers/unsigned/base.rs b/tfhe/src/high_level_api/integers/unsigned/base.rs
index 85a587743d..a2e8fefe7f 100644
--- a/tfhe/src/high_level_api/integers/unsigned/base.rs
+++ b/tfhe/src/high_level_api/integers/unsigned/base.rs
@@ -396,7 +396,7 @@ where
                     super::FheUint32Id::num_blocks(cuda_key.key.message_modulus),
                     streams,
                 );
-                super::FheUint32::new(result)
+                super::FheUint32::new(result, cuda_key.tag.clone())
             }),
         })
     }
@@ -440,7 +440,7 @@ where
                     super::FheUint32Id::num_blocks(cuda_key.key.message_modulus),
                     streams,
                 );
-                super::FheUint32::new(result)
+                super::FheUint32::new(result, cuda_key.tag.clone())
             }),
         })
     }
@@ -484,7 +484,7 @@ where
                     super::FheUint32Id::num_blocks(cuda_key.key.message_modulus),
                     streams,
                 );
-                super::FheUint32::new(result)
+                super::FheUint32::new(result, cuda_key.tag.clone())
             }),
         })
     }
@@ -528,7 +528,7 @@ where
                     super::FheUint32Id::num_blocks(cuda_key.key.message_modulus),
                     streams,
                 );
-                super::FheUint32::new(result)
+                super::FheUint32::new(result, cuda_key.tag.clone())
             }),
         })
     }
@@ -572,7 +572,7 @@ where
                     super::FheUint32Id::num_blocks(cuda_key.key.message_modulus),
                     streams,
                 );
-                super::FheUint32::new(result)
+                super::FheUint32::new(result, cuda_key.tag.clone())
             }),
         })
     }
@@ -625,7 +625,10 @@ where
                     super::FheUint32Id::num_blocks(cuda_key.key.message_modulus),
                     streams,
                 );
-                (super::FheUint32::new(result), FheBool::new(is_ok))
+                (
+                    super::FheUint32::new(result, cuda_key.tag.clone()),
+                    FheBool::new(is_ok, cuda_key.tag.clone()),
+                )
             }),
         })
     }