diff --git a/dockerimg/update_tests.sh b/dockerimg/update_tests.sh
index 5a42f7b..7127f3b 100755
--- a/dockerimg/update_tests.sh
+++ b/dockerimg/update_tests.sh
@@ -3,9 +3,8 @@
 SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
 CURRENT_DIR=$(pwd)
 
-if [ "$SCRIPT_DIR" != "$CURRENT_DIR" ]; then
-    echo "Error: This script must be run from its own directory: $SCRIPT_DIR" >&2
-    exit 1
-fi
+cd "$SCRIPT_DIR"
 
 go test -httprecord ".*" -rewritewant
+
+cd "$CURRENT_DIR"
diff --git a/loop/update_tests.sh b/loop/update_tests.sh
index f6be384..1e78edd 100755
--- a/loop/update_tests.sh
+++ b/loop/update_tests.sh
@@ -1,4 +1,11 @@
 #!/bin/bash
 
+SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
+CURRENT_DIR=$(pwd)
+
+cd "$SCRIPT_DIR"
+
 go test -httprecord .
 go test
+
+cd "$CURRENT_DIR"