diff --git a/install_incus_ollama.sh b/install_incus_ollama.sh
index 705deeb..b03ac4f 100644
--- a/install_incus_ollama.sh
+++ b/install_incus_ollama.sh
@@ -87,19 +87,11 @@ IMAGE="images:debian/12"
 
 
 #incus launch images:debian/12 ollama-container
-incus launch "${IMAGE}" "ollama-container" || die "No se pudo lanzar el contenedor"
+incus launch "${IMAGE}" "ollama-container" || die "${RED}No se pudo lanzar el contenedor${NC}"
 
-if [$? -eq 0]; then
-    echo -e "${GREEN}¡Ollama-container se ha inicializado correctamente!${NC}"
-else
-  
-   die "No se puede crear el container para Ollama"
+incus restart ollama-container || die "${RED}No se pudo reinciar el contenedor${NC}"
 
-fi
-
-incus restart ollama-container
-
-incus config device add ollama-container gpu gpu
+incus config device add ollama-container gpu gpu || die "${RED}No se pudo añadir la GPU al container${NC}"
 
 # Install nvidia