diff --git a/imap_processing/tests/ultra/unit/test_ultra_l1b_extended.py b/imap_processing/tests/ultra/unit/test_ultra_l1b_extended.py
index d3051e3a6..5d47796c4 100644
--- a/imap_processing/tests/ultra/unit/test_ultra_l1b_extended.py
+++ b/imap_processing/tests/ultra/unit/test_ultra_l1b_extended.py
@@ -288,7 +288,4 @@ def test_get_energy_pulse_height(de_dataset, yf_fixture):
     )
     test_energy = df_ph["Energy"].astype("float")
 
-    # TODO: change back when lut is added.
-    # assert np.array_equal(test_energy, energy[ph_indices])
-    energy[ph_indices] = test_energy
-    assert np.array_equal(test_energy, test_energy)
+    assert np.array_equal(test_energy, energy[ph_indices])
diff --git a/imap_processing/ultra/l1b/ultra_l1b_extended.py b/imap_processing/ultra/l1b/ultra_l1b_extended.py
index b6defecfd..e05879238 100644
--- a/imap_processing/ultra/l1b/ultra_l1b_extended.py
+++ b/imap_processing/ultra/l1b/ultra_l1b_extended.py
@@ -590,14 +590,14 @@ def get_energy_pulse_height(
     ylut[indices_bottom] = (yb[indices_bottom] / 100 + 82 / 2) * 32 / 82  # mm
 
     # TODO: waiting on these lookup tables: SpTpPHCorr, SpBtPHCorr
-    energy_ph[indices_top] = energy[
-        indices_top
-    ]  # - get_image_params("SpTpPHOffset") * SpTpPHCorr[
+    energy_ph[indices_top] = energy[indices_top] - get_image_params(
+        "SPTPPHOFF"
+    )  # * SpTpPHCorr[
     # xlut[indices_top], ylut[indices_top]] / 1024
 
-    energy_ph[indices_bottom] = energy[
-        indices_bottom
-    ]  # - get_image_params("SpBtPHOffset") * SpBtPHCorr[
+    energy_ph[indices_bottom] = energy[indices_bottom] - get_image_params(
+        "SPBTPHOFF"
+    )  # * SpBtPHCorr[
     # xlut[indices_bottom], ylut[indices_bottom]] / 1024
 
     return energy_ph