diff --git a/script/wrun b/script/wrun
index f75c9e17b306213c9c3709da0a09b0d0d64ba69e..61902c5f2c11e96a8e9420ae34b4aab891d90ba8 100755
--- a/script/wrun
+++ b/script/wrun
@@ -1,7 +1,5 @@
 #!/bin/bash
 
-#  -e DISPLAY=unix${DISPLAY}\
-
 if [[ $(lsb_release -rs) == "18.04" ]]; then
 docker run \
   --network host \
@@ -9,7 +7,7 @@ docker run \
   --ipc=host \
   -v /tmp/.X11-unix:/tmp/.X11-unix \
   -v /tmp:/exttmp \
-   -e DISPLAY=unix${DISPLAY}\
+  -e DISPLAY=unix${DISPLAY}\
   -e ROS_HOSTNAME=localhost \
   --rm -it --name waradevel gitlab.liu.se:5000/lrs/waraps_docker_images/waraps-melodic:devel "$@"
 fi
@@ -21,7 +19,7 @@ docker run \
   --ipc=host \
   -v /tmp/.X11-unix:/tmp/.X11-unix \
   -v /tmp:/exttmp \
-  -e DISPLAY=${DISPLAY}\
+  -e DISPLAY=unix${DISPLAY}\
   -e ROS_HOSTNAME=localhost \
   --rm -it --name waradevel gitlab.liu.se:5000/lrs/waraps_docker_images/waraps-noetic:devel "$@"
 fi