diff --git a/action/Dockerfile b/action/Dockerfile
index 4e33f8e..76e98d0 100644
--- a/action/Dockerfile
+++ b/action/Dockerfile
@@ -1,4 +1,5 @@
-FROM gableroux/unity3d:2019.2.11f1
+ARG IMAGE
+FROM $IMAGE
 
 LABEL "com.github.actions.name"="Unity - Activate"
 LABEL "com.github.actions.description"="Activate Unity using credentials or a license file. Both personal and professional licenses are supported."
diff --git a/action/entrypoint.sh b/action/entrypoint.sh
index 89bae1c..78843b0 100644
--- a/action/entrypoint.sh
+++ b/action/entrypoint.sh
@@ -1,5 +1,11 @@
 #!/usr/bin/env bash
 
+#
+# Display the unity version
+#
+
+echo "Activating Unity version \"$UNITY_VERSION\"."
+
 if [[ -n "$UNITY_LICENSE" ]]; then
   #
   # PERSONAL LICENSE MODE