diff --git a/figs/deep-retina-figures/f2_response.py b/figs/deep-retina-figures/f2_response.py
index 280465d..e204511 100644
--- a/figs/deep-retina-figures/f2_response.py
+++ b/figs/deep-retina-figures/f2_response.py
@@ -4,7 +4,7 @@
 import matplotlib.pyplot as plt
 from scipy.fftpack import fftfreq, fft
 from tqdm import tqdm
-from jetpack import errorplot
+from jetplot import errorplot
 import torchdeepretina as tdr
 import os
 import torch
diff --git a/torchdeepretina/retinal_phenomena.py b/torchdeepretina/retinal_phenomena.py
index 61d15a6..2f2157c 100644
--- a/torchdeepretina/retinal_phenomena.py
+++ b/torchdeepretina/retinal_phenomena.py
@@ -18,7 +18,7 @@
 from scipy.interpolate import interp1d
 from tqdm import tqdm
 try:
-    from jetpack import errorplot
+    from jetplot import errorplot
 except:
     print("f2_response is unavailable until you run:\n$ pip install -e git+git://github.com/nirum/jetpack.git@master#egg=jetpack")