diff --git a/Dockerfile b/Dockerfile
index e8d0ed8..4d3b648 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -70,7 +70,10 @@ COPY --from=train-env /root/venvs/train /root/venvs/train
 COPY --from=sia-env /root/venvs/sia /root/venvs/sia
 COPY --from=web-build /app/dist /root/static/
 
+RUN echo 'source /root/sia/scripts/add_venvs_to_path.sh' >> /root/.bashrc
+RUN echo 'source /root/sia/scripts/add_venvs_to_path.sh' >> /etc/profile
+
 WORKDIR /root/desktop
 
 ENTRYPOINT ["/bin/bash", "-c"]
-CMD ["/root/sia/scripts/install.sh; /bin/bash -lc /root/sia/scripts/restart.sh"]
+CMD ["/root/sia/scripts/install.sh; /bin/bash -c /root/sia/scripts/restart.sh"]
diff --git a/scripts/bootstrap.sh b/scripts/bootstrap.sh
index 0dea136..5505436 100755
--- a/scripts/bootstrap.sh
+++ b/scripts/bootstrap.sh
@@ -71,10 +71,10 @@ fi
 # Add venvs to path in current shell
 source /root/sia/scripts/add_venvs_to_path.sh
 
-## Finetune model
-#echo "Finetuning model..."
-#train
-#
-## Start SIA using restart script
-#echo "Run restart script..."
-#"/root/sia/scripts/restart.sh"
\ No newline at end of file
+# Finetune model
+echo "Finetuning model..."
+train
+
+# Start SIA using restart script
+echo "Run restart script..."
+"/root/sia/scripts/restart.sh"
\ No newline at end of file
diff --git a/scripts/exec.sh b/scripts/exec.sh
index ccf74a9..741619f 100755
--- a/scripts/exec.sh
+++ b/scripts/exec.sh
@@ -1,4 +1,4 @@
 #!/bin/bash
 
 container_id=$(docker ps -q)
-docker exec -it $container_id bash -l
\ No newline at end of file
+docker exec -it $container_id bash
\ No newline at end of file