diff --git a/action.yml b/action.yml
index 5cd355e..04e874d 100644
--- a/action.yml
+++ b/action.yml
@@ -32,7 +32,7 @@ inputs:
 
 runs:
   using: docker
-  image: docker://taptap21/docker-remote-deployment-action:v1.1
+  image: 'Dockerfile'
 
 
 branding:
diff --git a/docker-entrypoint.sh b/docker-entrypoint.sh
index a51573c..de47c3a 100755
--- a/docker-entrypoint.sh
+++ b/docker-entrypoint.sh
@@ -78,4 +78,6 @@ ${DEPLOYMENT_COMMAND} ${DEPLOYMENT_COMMAND_OPTIONS} pull
 echo "Command: ${DEPLOYMENT_COMMAND} ${INPUT_ARGS}"
 ${DEPLOYMENT_COMMAND} ${DEPLOYMENT_COMMAND_OPTIONS} ${INPUT_ARGS}
 
+echo "Remove docker context"
+docker context rm -f staging