diff --git a/applications/hmc/dwf/ensembleK.py b/applications/hmc/dwf/ensembleK.py
index d2742c49..097a0ad0 100755
--- a/applications/hmc/dwf/ensembleK.py
+++ b/applications/hmc/dwf/ensembleK.py
@@ -221,7 +221,7 @@ def mk_slv_s(solution_space, N, grid = None):
     m_l = 0.000302
     m_s = 0.0176
 
-else:
+elif latest_it <= 615:
     g.message("Use second-generation masses")
 
     # global fit estimate (mk)
@@ -235,6 +235,10 @@ def mk_slv_s(solution_space, N, grid = None):
     m_l = 0.000286
     m_s = 0.0169
 
+else:
+    g.message("Use third-generation masses")
+    m_l = 0.000262
+    m_s = 0.01643
 
 
 rat = g.algorithms.rational.zolotarev_inverse_square_root(1.0**0.5, 70.0**0.5, 11)