diff --git a/tests/constant_time/sig/passes/ml_dsa b/tests/constant_time/sig/passes/ml_dsa
index 69cc9ab23..ea9ff1f5b 100644
--- a/tests/constant_time/sig/passes/ml_dsa
+++ b/tests/constant_time/sig/passes/ml_dsa
@@ -18,53 +18,53 @@
    Memcheck:Cond
    fun:rej_gamma1m1
    fun:pqcrystals_ml_dsa*_ref_poly_uniform_gamma1m1
-   fun:crypto_sign_signature_ctx
+   fun:pqcrystals_ml_dsa*_ref_signature_internal
 }
 {
    Rejection sampling for challenge
    Memcheck:Cond
    fun:pqcrystals_ml_dsa*_ref_poly_challenge
-   fun:crypto_sign_signature_ctx
+   fun:pqcrystals_ml_dsa*_ref_signature_internal
 }
 {
    Rejection sampling for challenge
    Memcheck:Value8
    fun:pqcrystals_ml_dsa*_ref_poly_challenge
-   fun:crypto_sign_signature_ctx
+   fun:pqcrystals_ml_dsa*_ref_signature_internal
 }
 {
    Rejection sampling for signature distribution
    Memcheck:Cond
    ...
-   src:sign.c:166 # Call to polyvecl_chknorm
+   src:sign.c:161 # Call to polyvecl_chknorm
    # fun:pqcrystals_ml_dsa*_ref_signature
 }
 {
    Rejection sampling for signature distribution
    Memcheck:Cond
    ...
-   src:sign.c:175 # Call to polyveck_chknorm
+   src:sign.c:170 # Call to polyveck_chknorm
    # fun:pqcrystals_ml_dsa*_ref_signature
 }
 {
    Rejection sampling for signature distribution
    Memcheck:Cond
    ...
-   src:sign.c:182 # Call to polyveck_chknorm
+   src:sign.c:177 # Call to polyveck_chknorm
    # fun:pqcrystals_ml_dsa*_ref_signature
 }
 {
    Hint does not need to be computed in constant time
    Memcheck:Cond
    ...
-   src:sign.c:186 # Call to polyveck_make_hint
+   src:sign.c:181 # Call to polyveck_make_hint
    # fun:pqcrystals_ml_dsa*_ref_signature
 }
 {
    Rejection sampling for hint
    Memcheck:Cond
    ...
-   src:sign.c:187 # Checking number of 1 bits in hint
+   src:sign.c:182 # Checking number of 1 bits in hint
    # fun:pqcrystals_ml_dsa*_ref_signature
 }
 {
@@ -76,5 +76,5 @@
 {
    Verification is not done in constant time
    Memcheck:Cond
-   fun:pqcrystals_ml_dsa*_ref_verify
+   fun:pqcrystals_ml_dsa*_ref_verify_internal
 }
diff --git a/tests/constant_time/sig/passes/ml_dsa-avx2 b/tests/constant_time/sig/passes/ml_dsa-avx2
index d13da3234..3cb70dba2 100644
--- a/tests/constant_time/sig/passes/ml_dsa-avx2
+++ b/tests/constant_time/sig/passes/ml_dsa-avx2
@@ -34,14 +34,14 @@
    Memcheck:Cond
    ...
    fun:pqcrystals_ml_dsa*_avx2_poly_uniform_gamma1m1_4x
-   fun:crypto_sign_signature_ctx
+   fun:pqcrystals_ml_dsa*_avx2_signature_internal
 }
 {
    Rejection sampling for y
    Memcheck:Value8
    ...
    fun:pqcrystals_ml_dsa*_avx2_poly_uniform_gamma1m1_4x
-   fun:crypto_sign_signature_ctx
+   fun:pqcrystals_ml_dsa*_avx2_signature_internal
 }
 {
    Rejection sampling for s1 and s2
@@ -65,7 +65,7 @@
    ...
    fun:pqcrystals_ml_dsa*_avx2_poly_uniform_gamma1m1_preinit
    fun:pqcrystals_ml_dsa*_avx2_poly_uniform_gamma1m1
-   fun:crypto_sign_signature_ctx
+   fun:pqcrystals_ml_dsa*_avx2_signature_internal
 }
 {
    Rejection sampling for y
@@ -73,7 +73,7 @@
    ...
    fun:pqcrystals_ml_dsa*_avx2_poly_uniform_gamma1m1_preinit
    fun:pqcrystals_ml_dsa*_avx2_poly_uniform_gamma1m1
-   fun:crypto_sign_signature_ctx
+   fun:pqcrystals_ml_dsa*_avx2_signature_internal
 }
 {
    Rejection sampling for challenge
@@ -89,21 +89,21 @@
    Rejection sampling for signature distribution
    Memcheck:Cond
    ...
-   src:sign.c:253 # Call to poly_chknorm
+   src:sign.c:248 # Call to poly_chknorm
    # fun:pqcrystals_ml_dsa*_avx2_signature
 }
 {
    Rejection sampling for signature distribution
    Memcheck:Cond
    ...
-   src:sign.c:268 # Call to poly_chknorm
+   src:sign.c:263 # Call to poly_chknorm
    # fun:pqcrystals_ml_dsa*_avx2_signature
 }
 {
    Rejection sampling for signature distribution
    Memcheck:Cond
    ...
-   src:sign.c:275 # Call to poly_chknorm
+   src:sign.c:270 # Call to poly_chknorm
    # fun:pqcrystals_ml_dsa*_avx2_signature
 }
 {
@@ -111,34 +111,34 @@
    Memcheck:Cond
    ...
    fun:pqcrystals_ml_dsa*_avx2_poly_make_hint
-   src:sign.c:279 # fun:pqcrystals_ml_dsa*_ref_signature
+   src:sign.c:274 # fun:pqcrystals_ml_dsa*_avx2_signature
 }
 {
    Hint does not need to be computed in constant time
    Memcheck:Value8
    ...
    fun:pqcrystals_ml_dsa*_avx2_poly_make_hint
-   src:sign.c:279 # fun:pqcrystals_ml_dsa*_ref_signature
+   src:sign.c:274 # fun:pqcrystals_ml_dsa*_avx2_signature
 }
 {
    Rejection sampling for hint
    Memcheck:Cond
    ...
-   src:sign.c:280 # Checking number of 1 bits in hint
+   src:sign.c:275 # Checking number of 1 bits in hint
    # fun:pqcrystals_ml_dsa*_avx2_signature
 }
 {
    Hint positions are not secret
    Memcheck:Cond
    ...
-   src:sign.c:284 # memcpy
+   src:sign.c:279 # memcpy
    # fun:pqcrystals_ml_dsa*_avx2_signature
 }
 {
    Hint positions are not secret
    Memcheck:Value8
    ...
-   src:sign.c:284 # memcpy
+   src:sign.c:279 # memcpy
    # fun:pqcrystals_ml_dsa*_avx2_signature
 }
 {
@@ -150,6 +150,6 @@
 {
    Verification is not done in constant time
    Memcheck:Cond
-   fun:pqcrystals_ml_dsa*_avx2_verify
+   fun:pqcrystals_ml_dsa*_avx2_verify_internal
 }