diff --git a/files/HW-error-correction/hw.cxx b/files/HW-error-correction/hw.cxx
index ce14bc1..004a8aa 100644
--- a/files/HW-error-correction/hw.cxx
+++ b/files/HW-error-correction/hw.cxx
@@ -131,8 +131,8 @@ class HW : public PhysicsModel {
       client->run_model("hw_model_n", {inKeyN}, {outKeyN});
 
       // unpack output tensor
-      std::vector<double> correctionVort(n_values, 0);
-      std::vector<double> correctionN(n_values, 0);
+      std::vector<float> correctionVort(n_values, 0);
+      std::vector<float> correctionN(n_values, 0);
       client->unpack_tensor(outKeyVort, correctionVort.data(), {n_values},
                             SRTensorTypeFloat, SRMemLayoutContiguous);
       client->unpack_tensor(outKeyN, correctionN.data(), {n_values},