diff --git a/utils/aws/userdata.sh b/utils/aws/userdata.sh
index 36405d1a1565..a6d6e7976cf3 100644
--- a/utils/aws/userdata.sh
+++ b/utils/aws/userdata.sh
@@ -11,7 +11,8 @@ if [ ! -d yolov5 ]; then
   cd yolov5
   bash data/scripts/get_coco.sh && echo "Data done." &
   sudo docker pull ultralytics/yolov5:latest && echo "Docker done." &
-  # python -m pip install --upgrade pip && pip install -r requirements.txt && python detect.py && echo "Requirements done." &
+  python -m pip install --upgrade pip && pip install -r requirements.txt && python detect.py && echo "Requirements done." &
+  wait && echo "All tasks done." # finish background tasks
 else
   echo "Running re-start script." # resume interrupted runs
   i=0