diff --git a/files/5-training/submit-training.sh b/files/5-training/submit-training.sh
index c1a0216..1163d59 100644
--- a/files/5-training/submit-training.sh
+++ b/files/5-training/submit-training.sh
@@ -30,4 +30,8 @@ python ${SIMLINT_HOME}/files/5-training/training.py --epochs 100 --batch-size 32
 
 # model the density error
 python ${SIMLINT_HOME}/files/5-training/training.py --epochs 100 --batch-size 32 --learning-rate 0.0001 \
-    --trajectories 10 --data-directory ${WORK}/data/training/ --variables n --task-id dens
\ No newline at end of file
+    --trajectories 10 --data-directory ${WORK}/data/training/ --variables n --task-id dens
+
+# model the electrostatic potential error
+python ${SIMLINT_HOME}/files/5-training/training.py --epochs 100 --batch-size 32 --learning-rate 0.0001 \
+    --trajectories 10 --data-directory ${WORK}/data/training/ --variables phi --task-id phi
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},