diff --git a/src/tramdag/flow.py b/src/tramdag/flow.py
index 5601aa4..06fe455 100644
--- a/src/tramdag/flow.py
+++ b/src/tramdag/flow.py
@@ -629,7 +629,8 @@ def sample(self, n: int | None = None, *, do: dict[str, float] | None = None,
         values: dict[str, Tensor] = {}
         for name in self.order:
             if name in do:
-                values[name] = torch.full((n,), float(do[name]), device=self.device)
+                values[name] = torch.full((n,), float(do[name]),
+                                          dtype=self._dtype, device=self.device)
                 continue
             node = self.nodes[name]
             feats = self._features({p: values[p] for p in node.parents})