diff --git a/docker/entrypoint.sh b/docker/entrypoint.sh
index 989e663..6dc3763 100644
--- a/docker/entrypoint.sh
+++ b/docker/entrypoint.sh
@@ -9,11 +9,9 @@ CLONE_DIR="${CLONE_DIR:-/blog}"
 DRAFTS=""
 
 # Add drafts for preview
-if [ $BRANCH == "preview" ]; then
-  echo "- Adding draft pages for the site"
+if [ "$BRANCH" = "preview" ]; then
+  echo "- Adding draft pages to be generated"
   DRAFTS="--buildDrafts"
-else
-  echo "BRANCH= $BRANCH"
 fi
 
 # Clone repo