diff --git a/HISTORY.rst b/HISTORY.rst
index af16b736a7..5af35234b0 100644
--- a/HISTORY.rst
+++ b/HISTORY.rst
@@ -2,9 +2,12 @@
 History
 =======
 
-
 latest
 ------
+
+
+0.2.1 (2020-05-15)
+------
 * Add surface_precipitation_rate to one-step outputs, create training and test steps.
 * Correct prognostic runfile diagnostic calculations.
 * Update fv3gfs-python to v0.4.3.
@@ -14,7 +17,7 @@ latest
 * Multithreaded uploading in one-step jobs (#260)
 * Made nudging run upload more robust using k8s yaml templating submission and gsutil container upload
 * The key for commands in the end to end config YAML can be given as either `command` or `argo`, and the arguments will be parsed into the appropriate format for either type.
-
+* Offline diags workflow now downloads the test data to a local temp dir to speed reading and prevent remote read errors.
 
 0.2.0 (2020-04-23)
 ------------------
diff --git a/fv3net/__init__.py b/fv3net/__init__.py
index c7ff07f0bb..018860790a 100644
--- a/fv3net/__init__.py
+++ b/fv3net/__init__.py
@@ -1,7 +1,7 @@
 import pathlib
 
 __author__ = """Vulcan Technologies LLC"""
-__version__ = "0.2.0"
+__version__ = "0.2.1"
 
 TOP_LEVEL_DIR = pathlib.Path(__file__).parent.parent.absolute()
 
diff --git a/setup.cfg b/setup.cfg
index 849e526621..1da4983b9c 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 0.2.0
+current_version = 0.2.1
 commit = True
 tag = False
 
@@ -19,3 +19,4 @@ replace = fv3net:v{new_version}
 exclude = docs
 ignore = E203,W293,W503
 max-line-length = 88
+
diff --git a/setup.py b/setup.py
index 7037f08f5a..d24afa030d 100644
--- a/setup.py
+++ b/setup.py
@@ -27,7 +27,7 @@
     packages=packages,
     install_requires=dependencies,
     package_dir=package_dir,
-    version="0.2.0",
+    version="0.2.1",
     description="Improving the GFDL FV3 model physics with machine learning",
     author="Vulcan Inc.",
     license="MIT",
diff --git a/workflows/prognostic_run_diags/argo.yaml b/workflows/prognostic_run_diags/argo.yaml
index 423e41b075..935aa820b9 100644
--- a/workflows/prognostic_run_diags/argo.yaml
+++ b/workflows/prognostic_run_diags/argo.yaml
@@ -32,7 +32,7 @@ spec:
        value: "climate-sim-pool"
        effect: "NoSchedule"
      container:
-      image: us.gcr.io/vcm-ml/fv3net:eea56be96af7ec828a62cf8f3400f521e42c1ba1
+      image: us.gcr.io/vcm-ml/fv3net:v0.2.1
       command:
       - python
       - generate_report.py
@@ -70,7 +70,7 @@ spec:
        value: "climate-sim-pool"
        effect: "NoSchedule"
      container:
-      image: us.gcr.io/vcm-ml/fv3net:eea56be96af7ec828a62cf8f3400f521e42c1ba1
+      image: us.gcr.io/vcm-ml/fv3net:v0.2.1
       command: ["bash", "-x", "-e"]
       args:
       - entrypoint.sh