diff --git a/quartical/gains/amplitude/kernel.py b/quartical/gains/amplitude/kernel.py
index 2f37b752..b197d601 100644
--- a/quartical/gains/amplitude/kernel.py
+++ b/quartical/gains/amplitude/kernel.py
@@ -227,6 +227,8 @@ def nb_compute_jhj_jhr(
     corr_mode
 ):
 
+    coerce_literal(nb_compute_jhj_jhr, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     row_weights_idx = ms_inputs.fields.index('ROW_WEIGHTS')
     row_weights_type = ms_inputs[row_weights_idx]
@@ -449,6 +451,8 @@ def compute_update(native_imdry, corr_mode):
 @overload(compute_update, jit_options=PARALLEL_JIT_OPTIONS)
 def nb_compute_update(native_imdry, corr_mode):
 
+    coerce_literal(nb_compute_update, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     jhj = native_imdry[native_imdry.fields.index('jhj')]
 
@@ -505,6 +509,8 @@ def nb_finalize_update(
     corr_mode
 ):
 
+    coerce_literal(nb_finalize_update, ["corr_mode"])
+
     set_identity = factories.set_identity_factory(corr_mode)
     param_to_gain = param_to_gain_factory(corr_mode)
 
diff --git a/quartical/gains/complex/diag_kernel.py b/quartical/gains/complex/diag_kernel.py
index dc45c38e..8bd6ba59 100644
--- a/quartical/gains/complex/diag_kernel.py
+++ b/quartical/gains/complex/diag_kernel.py
@@ -206,6 +206,8 @@ def nb_compute_jhj_jhr(
     corr_mode
 ):
 
+    coerce_literal(nb_compute_jhj_jhr, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     row_weights_idx = ms_inputs.fields.index('ROW_WEIGHTS')
     row_weights_type = ms_inputs[row_weights_idx]
@@ -428,6 +430,8 @@ def compute_update(native_imdry, corr_mode):
 @overload(compute_update, jit_options=PARALLEL_JIT_OPTIONS)
 def nb_compute_update(native_imdry, corr_mode):
 
+    coerce_literal(nb_compute_update, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     jhj = native_imdry[native_imdry.fields.index('jhj')]
 
@@ -484,6 +488,8 @@ def nb_finalize_update(
     corr_mode
 ):
 
+    coerce_literal(nb_finalize_update, ["corr_mode"])
+
     set_identity = factories.set_identity_factory(corr_mode)
 
     def impl(
diff --git a/quartical/gains/complex/kernel.py b/quartical/gains/complex/kernel.py
index 446a946a..f8ca97dc 100644
--- a/quartical/gains/complex/kernel.py
+++ b/quartical/gains/complex/kernel.py
@@ -207,6 +207,8 @@ def nb_compute_jhj_jhr(
     corr_mode
 ):
 
+    coerce_literal(nb_compute_jhj_jhr, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     row_weights_idx = ms_inputs.fields.index('ROW_WEIGHTS')
     row_weights_type = ms_inputs[row_weights_idx]
@@ -429,6 +431,8 @@ def compute_update(native_imdry, corr_mode):
 @overload(compute_update, jit_options=PARALLEL_JIT_OPTIONS)
 def nb_compute_update(native_imdry, corr_mode):
 
+    coerce_literal(nb_compute_update, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     jhj = native_imdry[native_imdry.fields.index('jhj')]
 
@@ -485,6 +489,8 @@ def nb_finalize_update(
     corr_mode
 ):
 
+    coerce_literal(nb_finalize_update, ["corr_mode"])
+
     set_identity = factories.set_identity_factory(corr_mode)
 
     def impl(
diff --git a/quartical/gains/crosshand_phase/kernel.py b/quartical/gains/crosshand_phase/kernel.py
index 74fb5678..53adfff4 100644
--- a/quartical/gains/crosshand_phase/kernel.py
+++ b/quartical/gains/crosshand_phase/kernel.py
@@ -225,6 +225,8 @@ def nb_compute_jhj_jhr(
     corr_mode
 ):
 
+    coerce_literal(nb_compute_jhj_jhr, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     row_weights_idx = ms_inputs.fields.index('ROW_WEIGHTS')
     row_weights_type = ms_inputs[row_weights_idx]
@@ -453,6 +455,8 @@ def compute_update(native_imdry, corr_mode):
 @overload(compute_update, jit_options=PARALLEL_JIT_OPTIONS)
 def nb_compute_update(native_imdry, corr_mode):
 
+    coerce_literal(nb_compute_update, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     jhj = native_imdry[native_imdry.fields.index('jhj')]
 
@@ -509,6 +513,8 @@ def nb_finalize_update(
     corr_mode
 ):
 
+    coerce_literal(nb_finalize_update, ["corr_mode"])
+
     set_identity = factories.set_identity_factory(corr_mode)
     param_to_gain = param_to_gain_factory(corr_mode)
 
diff --git a/quartical/gains/delay/kernel.py b/quartical/gains/delay/kernel.py
index d95efa02..209da746 100644
--- a/quartical/gains/delay/kernel.py
+++ b/quartical/gains/delay/kernel.py
@@ -254,6 +254,8 @@ def nb_compute_jhj_jhr(
     corr_mode
 ):
 
+    coerce_literal(nb_compute_jhj_jhr, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     row_weights_idx = ms_inputs.fields.index('ROW_WEIGHTS')
     row_weights_type = ms_inputs[row_weights_idx]
@@ -493,6 +495,8 @@ def compute_update(native_imdry, corr_mode):
 @overload(compute_update, jit_options=PARALLEL_JIT_OPTIONS)
 def nb_compute_update(native_imdry, corr_mode):
 
+    coerce_literal(nb_compute_update, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     jhj = native_imdry[native_imdry.fields.index('jhj')]
 
@@ -553,6 +557,8 @@ def nb_finalize_update(
     corr_mode
 ):
 
+    coerce_literal(nb_finalize_update, ["corr_mode"])
+
     set_identity = factories.set_identity_factory(corr_mode)
     param_to_gain = param_to_gain_factory(corr_mode)
 
diff --git a/quartical/gains/delay_and_offset/kernel.py b/quartical/gains/delay_and_offset/kernel.py
index eebc09e1..70e41d95 100644
--- a/quartical/gains/delay_and_offset/kernel.py
+++ b/quartical/gains/delay_and_offset/kernel.py
@@ -248,6 +248,8 @@ def nb_compute_jhj_jhr(
     corr_mode
 ):
 
+    coerce_literal(nb_compute_jhj_jhr, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     row_weights_idx = ms_inputs.fields.index('ROW_WEIGHTS')
     row_weights_type = ms_inputs[row_weights_idx]
@@ -487,6 +489,8 @@ def compute_update(native_imdry, corr_mode):
 @overload(compute_update, jit_options=PARALLEL_JIT_OPTIONS)
 def nb_compute_update(native_imdry, corr_mode):
 
+    coerce_literal(nb_compute_update, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     jhj = native_imdry[native_imdry.fields.index('jhj')]
 
@@ -547,6 +551,8 @@ def nb_finalize_update(
     corr_mode
 ):
 
+    coerce_literal(nb_finalize_update, ["corr_mode"])
+
     set_identity = factories.set_identity_factory(corr_mode)
     param_to_gain = param_to_gain_factory(corr_mode)
 
diff --git a/quartical/gains/leakage/kernel.py b/quartical/gains/leakage/kernel.py
index b9cbc6e6..8fa6921a 100644
--- a/quartical/gains/leakage/kernel.py
+++ b/quartical/gains/leakage/kernel.py
@@ -200,6 +200,8 @@ def nb_finalize_update(
     corr_mode
 ):
 
+    coerce_literal(nb_finalize_update, ["corr_mode"])
+
     set_identity = factories.set_identity_factory(corr_mode)
 
     def impl(
diff --git a/quartical/gains/phase/kernel.py b/quartical/gains/phase/kernel.py
index 50a03cc5..7fa76da0 100644
--- a/quartical/gains/phase/kernel.py
+++ b/quartical/gains/phase/kernel.py
@@ -232,6 +232,8 @@ def nb_compute_jhj_jhr(
     corr_mode
 ):
 
+    coerce_literal(nb_compute_jhj_jhr, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     row_weights_idx = ms_inputs.fields.index('ROW_WEIGHTS')
     row_weights_type = ms_inputs[row_weights_idx]
@@ -460,6 +462,8 @@ def compute_update(native_imdry, corr_mode):
 @overload(compute_update, jit_options=PARALLEL_JIT_OPTIONS)
 def nb_compute_update(native_imdry, corr_mode):
 
+    coerce_literal(nb_compute_update, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     jhj = native_imdry[native_imdry.fields.index('jhj')]
 
@@ -516,6 +520,8 @@ def nb_finalize_update(
     corr_mode
 ):
 
+    coerce_literal(nb_finalize_update, ["corr_mode"])
+
     set_identity = factories.set_identity_factory(corr_mode)
     param_to_gain = param_to_gain_factory(corr_mode)
 
diff --git a/quartical/gains/rotation/kernel.py b/quartical/gains/rotation/kernel.py
index f18fca70..66cc58c2 100644
--- a/quartical/gains/rotation/kernel.py
+++ b/quartical/gains/rotation/kernel.py
@@ -228,6 +228,8 @@ def nb_compute_jhj_jhr(
     corr_mode
 ):
 
+    coerce_literal(nb_compute_jhj_jhr, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     row_weights_idx = ms_inputs.fields.index('ROW_WEIGHTS')
     row_weights_type = ms_inputs[row_weights_idx]
@@ -463,6 +465,8 @@ def compute_update(native_imdry, corr_mode):
 @overload(compute_update, jit_options=PARALLEL_JIT_OPTIONS)
 def nb_compute_update(native_imdry, corr_mode):
 
+    coerce_literal(nb_compute_update, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     jhj = native_imdry[native_imdry.fields.index('jhj')]
 
@@ -521,6 +525,8 @@ def nb_finalize_update(
     corr_mode
 ):
 
+    coerce_literal(nb_finalize_update, ["corr_mode"])
+
     set_identity = factories.set_identity_factory(corr_mode)
 
     if corr_mode.literal_value == 4:
diff --git a/quartical/gains/rotation_measure/kernel.py b/quartical/gains/rotation_measure/kernel.py
index 8dc2dc5f..3c935f23 100644
--- a/quartical/gains/rotation_measure/kernel.py
+++ b/quartical/gains/rotation_measure/kernel.py
@@ -234,6 +234,8 @@ def nb_compute_jhj_jhr(
     corr_mode
 ):
 
+    coerce_literal(nb_compute_jhj_jhr, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     row_weights_idx = ms_inputs.fields.index('ROW_WEIGHTS')
     row_weights_type = ms_inputs[row_weights_idx]
@@ -474,6 +476,8 @@ def compute_update(native_imdry, corr_mode):
 @overload(compute_update, jit_options=PARALLEL_JIT_OPTIONS)
 def nb_compute_update(native_imdry, corr_mode):
 
+    coerce_literal(nb_compute_update, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     jhj = native_imdry[native_imdry.fields.index('jhj')]
 
@@ -534,6 +538,8 @@ def nb_finalize_update(
     corr_mode
 ):
 
+    coerce_literal(nb_finalize_update, ["corr_mode"])
+
     set_identity = factories.set_identity_factory(corr_mode)
 
     if corr_mode.literal_value == 4:
diff --git a/quartical/gains/tec_and_offset/kernel.py b/quartical/gains/tec_and_offset/kernel.py
index 155ac67e..dd3d95ac 100644
--- a/quartical/gains/tec_and_offset/kernel.py
+++ b/quartical/gains/tec_and_offset/kernel.py
@@ -250,6 +250,8 @@ def nb_compute_jhj_jhr(
     corr_mode
 ):
 
+    coerce_literal(nb_compute_jhj_jhr, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     row_weights_idx = ms_inputs.fields.index('ROW_WEIGHTS')
     row_weights_type = ms_inputs[row_weights_idx]
@@ -490,6 +492,8 @@ def compute_update(native_imdry, corr_mode):
 @overload(compute_update, jit_options=PARALLEL_JIT_OPTIONS)
 def nb_compute_update(native_imdry, corr_mode):
 
+    coerce_literal(nb_compute_update, ["corr_mode"])
+
     # We want to dispatch based on this field so we need its type.
     jhj = native_imdry[native_imdry.fields.index('jhj')]
 
@@ -550,6 +554,8 @@ def nb_finalize_update(
     corr_mode
 ):
 
+    coerce_literal(nb_finalize_update, ["corr_mode"])
+
     set_identity = factories.set_identity_factory(corr_mode)
     param_to_gain = param_to_gain_factory(corr_mode)