diff --git a/integration-test.sh b/integration-test.sh
index cb796ca..545c3c7 100755
--- a/integration-test.sh
+++ b/integration-test.sh
@@ -6,6 +6,9 @@
 
 set -ex
 
+export SPORESTACK_ENDPOINT=https://api.sporestack.com
+# export SPORESTACK_ENDPOINT=http://127.0.0.1:8000
+
 export SPORESTACK_DIR=$(pwd)/dummydotsporestackfolder
 
 rm -r $SPORESTACK_DIR
@@ -15,7 +18,7 @@ sporestack version
 sporestack version | grep '[0-9]\.[0-9]\.[0-9]'
 
 sporestack api-endpoint
-sporestack api-endpoint | grep api.sporestack.com
+sporestack api-endpoint | grep "$SPORESTACK_ENDPOINT"
 
 sporestack token list
 sporestack token list 2>&1 | wc -l | grep '2$'
@@ -34,6 +37,7 @@ sporestack server flavors | grep vcpu
 sporestack server operating-systems | grep debian-11
 
 if [ -z "$REAL_TESTING_TOKEN" ]; then
+	rm -r $SPORESTACK_DIR
 	echo "REAL_TESTING_TOKEN not set, not finishing tests."
 	echo Success
 	exit 0
@@ -61,4 +65,6 @@ sporestack server rebuild --token realtestingtoken --hostname sporestackpythonin
 sporestack server delete --token realtestingtoken --hostname sporestackpythonintegrationtestdelme
 sporestack server forget --token realtestingtoken --hostname sporestackpythonintegrationtestdelme
 
+rm -r $SPORESTACK_DIR
+
 echo Success