diff --git a/.circleci/config.yml b/.circleci/config.yml
index df8133aa..e22b8cb3 100644
--- a/.circleci/config.yml
+++ b/.circleci/config.yml
@@ -56,7 +56,7 @@ setup_ptv_conda: &setup_ptv_conda
       conda update -y conda
       conda init bash
       source ~/.bashrc
-      conda create --name pytorchvideo python=3.7.9
+      conda create --name pytorchvideo python=3.8.13
 
 install_pytorch: &install_pytorch
   - run:
diff --git a/pytorchvideo/layers/attention.py b/pytorchvideo/layers/attention.py
index bfcf61ef..f6fdc739 100644
--- a/pytorchvideo/layers/attention.py
+++ b/pytorchvideo/layers/attention.py
@@ -4,6 +4,7 @@
 
 import numpy
 import torch
+import torch.fx
 import torch.nn as nn
 from torch.nn.common_types import _size_3_t
 
diff --git a/setup.py b/setup.py
index abe7eefc..b4fffca8 100755
--- a/setup.py
+++ b/setup.py
@@ -52,7 +52,7 @@ def get_name():
     python_requires=">=3.7",
     install_requires=[
         "fvcore",
-        "av",
+        "av==8.0.3",
         "parameterized",
         "iopath",
         "networkx",