diff --git a/examples/o3-averaging/o3-averaging.py b/examples/o3-averaging/o3-averaging.py
index d7eb9226..a3b30c9f 100644
--- a/examples/o3-averaging/o3-averaging.py
+++ b/examples/o3-averaging/o3-averaging.py
@@ -11,6 +11,7 @@
 
 # %%
 
+import bz2
 import os
 import subprocess
 import time
@@ -24,6 +25,8 @@
 
 # import matplotlib.pyplot as plt
 # import numpy as np
+import matplotlib.pyplot as plt
+import numpy as np
 
 
 # %%