diff --git a/scripts/diff.sh b/scripts/diff.sh
index 521606d..196402e 100755
--- a/scripts/diff.sh
+++ b/scripts/diff.sh
@@ -28,10 +28,14 @@ if [[ ! -f "$migration" ]]; then
   
   echo "migrate from $rev => HEAD" 1>&2
 
-  echo "BEGIN" > "$migration"
+  echo "do language plpgsql \$migration\$" >> "$migration"
+  echo "begin" >> "$migration"
+  echo >> "$migration"
   migra --unsafe $base_url $head_url >> "$migration" \
     || echo "migra exited with code $?. this is /probably/ fine" 1>&2
-  echo "COMMIT;" > "$migration"
+  echo >> "$migration"
+  echo "end;" >> "$migration"
+  echo '$migration$;' >> "$migration"
 fi
 
 echo "$migration"