diff --git a/setup/requirements.txt b/setup/requirements.txt
index c69143dd6..d8e64ad1d 100644
--- a/setup/requirements.txt
+++ b/setup/requirements.txt
@@ -1,5 +1,5 @@
+cython<3.0.0 # pyyaml is broken with cython 3
 jsonlines
 megatron_core>=0.8
 nemo_toolkit[nlp]
 nvidia-pytriton
-cython<3.0.0 # pyyaml is broken with cython 3