diff --git a/.github/workflows/trymerge.yml b/.github/workflows/trymerge.yml
index 876c62e3d5..699acd7d9c 100644
--- a/.github/workflows/trymerge.yml
+++ b/.github/workflows/trymerge.yml
@@ -26,6 +26,8 @@ jobs:
           check-latest: false
           cache: pip
           architecture: x64
+      # TODO (huydhn): get rid of Rockset
+      - run: pip install pyyaml==6.0 rockset==1.0.3
 
       - name: Setup committer id
         run: |