diff --git a/entrypoint.sh b/entrypoint.sh
index 5a3ad17..c1074fa 100755
--- a/entrypoint.sh
+++ b/entrypoint.sh
@@ -14,4 +14,6 @@ x11vnc -display :0 -forever -ncache 10 &
 # Add delay
 sleep 5
 
+# Activate virtual environment
+export DISPLAY=:0
 . .venv/bin/activate && python3 main.py
\ No newline at end of file