diff --git a/lib_test/test_client_connection.ml b/lib_test/test_client_connection.ml
index ae0ad927..70564cd2 100644
--- a/lib_test/test_client_connection.ml
+++ b/lib_test/test_client_connection.ml
@@ -41,7 +41,7 @@ let read_response t r =
 
 let reader_ready t =
   Alcotest.check read_operation "Reader is ready"
-    `Read (next_read_operation t :> [`Close | `Read | `Yield | `Upgrade]);
+    `Read (next_read_operation t :> Read_operation.t);
 ;;
 
 let write_string ?(msg="output written") t str =
@@ -69,7 +69,7 @@ let writer_closed t =
 
 let connection_is_shutdown t =
   Alcotest.check read_operation "Reader is closed"
-    `Close (next_read_operation t :> [`Close | `Read | `Yield | `Upgrade]);
+    `Close (next_read_operation t :> Read_operation.t);
   writer_closed t;
 ;;