diff --git a/src/enzyme_ad/jax/Passes/EnzymeHLOOpt.cpp b/src/enzyme_ad/jax/Passes/EnzymeHLOOpt.cpp
index 661bd7fb7..ec0cf0f7e 100644
--- a/src/enzyme_ad/jax/Passes/EnzymeHLOOpt.cpp
+++ b/src/enzyme_ad/jax/Passes/EnzymeHLOOpt.cpp
@@ -267,6 +267,7 @@ DenseElementsAttr fromTensor(stablehlo::Tensor inp) {
   }
 
   assert(0);
+  return {};
 }
 
 /*
diff --git a/src/enzyme_ad/jax/primitives.py b/src/enzyme_ad/jax/primitives.py
index d7c93b630..bb1e19b2f 100644
--- a/src/enzyme_ad/jax/primitives.py
+++ b/src/enzyme_ad/jax/primitives.py
@@ -462,6 +462,7 @@ def maketup(ty):
     ty = ir.RankedTensorType(ty)
     tystr = ty.element_type.__str__()
     tystr = {
+        "bf16": "bfloat16",
         "f32": "float",
         "f64": "double",
         "i32": "int32_t",