diff --git a/docker-compose.yml b/docker-compose.yml
index 455e6fe..a36c6d6 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -19,3 +19,4 @@ services:
       # Difficulty settings: 0 - peaceful, 1 - easy, 2 - normal, 3 - hard
       DIFFICULTY: "easy"
       MOTD: "A Dockerized Minecraft Server"
+      WORLD_TYPE: "default"
diff --git a/setup.sh b/setup.sh
index 2119d81..11839f6 100755
--- a/setup.sh
+++ b/setup.sh
@@ -90,5 +90,11 @@ if [[ -n $MOTD ]]; then
   echo "motd=$MOTD" >> $SERVER_PROPERTIES
 fi
 
+if [[ -n $WORLD_TYPE ]]; then
+  echo "world-type=$WORLD_TYPE" >> $SERVER_PROPERTIES
+fi
+
+sleep 1
+
 # Start server
-exec java $JAVA_OPTS -jar "server.jar" nogui
\ No newline at end of file
+exec java $JAVA_OPTS -jar "server.jar" nogui