diff --git a/entrypoint.sh b/entrypoint.sh
index d60d2e0..11ef38b 100644
--- a/entrypoint.sh
+++ b/entrypoint.sh
@@ -6,4 +6,15 @@ echo "Launching command:"
 echo "$@"
 echo "----------"
 
-"$@"
+CMD=$*
+
+if [ "${CMD}" == "" ]; then
+    CMD="${INPUT_COMMAND}"
+fi
+
+if [ "${CMD}" == "" ]; then
+    echo "! No command to run!"
+    exit 1
+fi
+
+"${CMD}"