1 files changed, 3 insertions, 1 deletions
diff --git a/check b/check
index cde86a6..bc57b7a 100755
--- a/check
+++ b/check
@@ -4,4 +4,6 @@ set -eu
 
 rm -f test.py
 black --check contractor *.py
-CONTRACTOR_ADDRESS="$1" sp-codegen contractor.md -o test.py --run
+echo "$1" > test.address
+echo "SSH_AUTH_SOCK=$SSH_AUTH_SOCK" > test.env
+sp-codegen contractor.md -o test.py --run