diff --git a/Dockerfile-cuda b/Dockerfile-cuda
index 393f8c2..684374e 100644
--- a/Dockerfile-cuda
+++ b/Dockerfile-cuda
@@ -15,9 +15,6 @@ run apt-get update -qq && apt-get install -qqy \
     
 run curl -sSL https://get.docker.com/ | sh
 
-add ./wrapdocker /usr/local/bin/wrapdocker
-run chmod +x /usr/local/bin/wrapdocker
-
 volume /var/lib/docker
 
 env HF_HOME hf_home
@@ -33,4 +30,7 @@ run pip install -r requirements.cuda.txt
 run pip install -r requirements.txt
 run pip install -e .
 
-cmd ["wrapdocker"]
+add ./wrapdocker /usr/local/bin/wrapdocker
+run chmod +x /usr/local/bin/wrapdocker
+
+cmd ["wrapdocker", "bash"]
diff --git a/skynet/dgpu/network.py b/skynet/dgpu/network.py
index 5309122..3a30727 100644
--- a/skynet/dgpu/network.py
+++ b/skynet/dgpu/network.py
@@ -27,7 +27,8 @@ async def failable(fn: partial, ret_fail=None):
 
     except (
         asks.errors.RequestTimeout,
-        json.JSONDecodeError
+        json.JSONDecodeError,
+        OSError
     ):
         return ret_fail
 
diff --git a/wrapdocker b/wrapdocker
index ac927a3..0d74508 100755
--- a/wrapdocker
+++ b/wrapdocker
@@ -108,6 +108,5 @@ else
 		fi
 		sleep 1
 	done
-	[[ $1 ]] && exec "$@"
-	exec bash --login
+	[[ $1 ]] && exec bash -c "$@"
 fi