diff --git a/.github/workflows/cml.yaml b/.github/workflows/cml.yaml
index fa5e65d..e4c622e 100644
--- a/.github/workflows/cml.yaml
+++ b/.github/workflows/cml.yaml
@@ -4,6 +4,9 @@ jobs:
   train-model:
     runs-on: ubuntu-latest
     steps:
+      - uses: actions/setup-node@v1
+        with:
+          node-version: '16'
       - uses: actions/checkout@v3
         with:
           ref: ${{ github.event.pull_request.head.sha }}