diff --git a/.github/workflows/pr-a10g.yml b/.github/workflows/pr-a10g.yml
index c5f549443d..fb390a7638 100644
--- a/.github/workflows/pr-a10g.yml
+++ b/.github/workflows/pr-a10g.yml
@@ -52,6 +52,7 @@ jobs:
             tail -f /dev/null
           )
           echo "Container name: ${container_name}"
+          docker exec -t -w "/" "${container_name}" bash -c "sudo chown -R runner /benchmark; sudo chgrp -R runner /benchmark"
           docker exec -t -w "/benchmark" "${container_name}" bash /benchmark/scripts/torchbench_install.sh
           docker exec -t -w "/benchmark" "${container_name}" bash /benchmark/scripts/torchbench_test.sh
       - name: Teardown Linux
diff --git a/torchbenchmark/models/BERT_pytorch/install.py b/torchbenchmark/models/BERT_pytorch/install.py
index 26ab44a808..f1533e07a6 100644
--- a/torchbenchmark/models/BERT_pytorch/install.py
+++ b/torchbenchmark/models/BERT_pytorch/install.py
@@ -3,7 +3,7 @@
 
 
 def setup_install():
-    subprocess.check_call([sys.executable, 'setup.py', 'develop'])
+    subprocess.check_call([sys.executable, '-m', "pip", 'install', '--user', '-e', '.'])
 
 if __name__ == '__main__':
     setup_install()