diff --git a/Dockerfile b/Dockerfile
index a512090..6a902c9 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -113,7 +113,7 @@ RUN set -xe; \
     mv /tmp/alpine/bin/* /usr/local/bin; \
     \
     { \
-        echo 'export PS1="\u@${WODBY_APP_NAME:-python}.${WODBY_ENVIRONMENT_NAME:-container}:\w $ "'; \
+        echo 'export PS1="\u@$(hostname):\w $ "'; \
         echo "export PATH=${PATH}"; \
     } | tee /home/wodby/.shrc; \
     \
diff --git a/tests/tests.sh b/tests/tests.sh
index 2e07600..38ad994 100755
--- a/tests/tests.sh
+++ b/tests/tests.sh
@@ -6,6 +6,13 @@ if [[ -n "${DEBUG}" ]]; then
     set -x
 fi
 
+source /home/wodby/.shrc
+expected_prompt='\u@'"$(hostname)"':\w $ '
+if [[ "${PS1}" != "${expected_prompt}" ]]; then
+    echo "Unexpected shell prompt: ${PS1}"
+    exit 1
+fi
+
 if [[ "${PYTHON_VERSION}" == 2* ]]; then
     python -V 2>&1 | grep -q "${PYTHON_VERSION}"
     django_msg="It worked!"