[FLINK-16838] Update base image version tag in update_branch_version.sh

This closes #78.
diff --git a/tools/docker/build-stateful-functions.sh b/tools/docker/build-stateful-functions.sh
index f4cddf7..08b3847 100755
--- a/tools/docker/build-stateful-functions.sh
+++ b/tools/docker/build-stateful-functions.sh
@@ -17,6 +17,10 @@
 
 set -e
 
+#
+# Do not change the name of this variable;
+# it is referenced in the tools/releasing/update_branch_version.sh script
+#
 VERSION_TAG=2.1-SNAPSHOT
 
 #
diff --git a/tools/releasing/update_branch_version.sh b/tools/releasing/update_branch_version.sh
index 95cf897..26552b2 100755
--- a/tools/releasing/update_branch_version.sh
+++ b/tools/releasing/update_branch_version.sh
@@ -73,6 +73,10 @@
 perl -pi -e "s#version: \"$OLD_VERSION\"#version: \"$NEW_VERSION\"#" docs/_config.yml
 perl -pi -e "s#version_title: \"$OLD_VERSION\"#version_title: \"$NEW_VERSION\"#" docs/_config.yml
 
+# change Stateful Functions image version tags in all Dockerfiles and image build script
+find . -name 'Dockerfile*' -type f -exec perl -pi -e "s#FROM flink-statefun:$OLD_VERSION#FROM flink-statefun:$NEW_VERSION#" {} \;
+perl -pi -e "s#VERSION_TAG=$OLD_VERSION#VERSION_TAG=$NEW_VERSION#" tools/docker/build-stateful-functions.sh
+
 git commit -am "[release] Update version to ${NEW_VERSION}"
 
 NEW_VERSION_COMMIT_HASH=`git rev-parse HEAD`