diff --git a/cog.yaml b/cog.yaml
index 0541f8d..9d2e90b 100644
--- a/cog.yaml
+++ b/cog.yaml
@@ -6,7 +6,7 @@ build:
     - "libgl1-mesa-glx"
     - "libglib2.0-0"
   python_packages:
-    - "min-dalle==0.3.4"
+    - "min-dalle==0.3.5"
   run:
     - pip install torch==1.11.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html
 
diff --git a/min_dalle/min_dalle.py b/min_dalle/min_dalle.py
index 686659e..600b319 100644
--- a/min_dalle/min_dalle.py
+++ b/min_dalle/min_dalle.py
@@ -1,6 +1,5 @@
 import os
 from PIL import Image
-from matplotlib.pyplot import grid
 import numpy
 from torch import LongTensor, FloatTensor
 from math import sqrt
diff --git a/setup.py b/setup.py
index 9675cb7..0c8b658 100644
--- a/setup.py
+++ b/setup.py
@@ -5,7 +5,7 @@ setuptools.setup(
     name='min-dalle',
     description = 'min(DALL·E)',
     # long_description=(Path(__file__).parent / "README.rst").read_text(),
-    version='0.3.4',
+    version='0.3.5',
     author='Brett Kuprel',
     author_email='brkuprel@gmail.com',
     url='https://github.com/kuprel/min-dalle',