diff --git a/prover/src/epoch_service.rs b/prover/src/epoch_service.rs
index dfc1285..bdc6f14 100644
--- a/prover/src/epoch_service.rs
+++ b/prover/src/epoch_service.rs
@@ -389,7 +389,7 @@ mod tests {
             let res = epoch_service.compute_wait_until(&now, &now);
 
             // expect TooLow error
-            assert!(matches!(res, Err(EpochServiceError::TooLow(_, _))));
+            assert!(matches!(res, Err(EpochServiceError::WaitUntilTooLow(_, _))));
         }
     }