diff --git a/PanACoTA/annotate_module/annotation_functions.py b/PanACoTA/annotate_module/annotation_functions.py
index 81846f0f56808c93c45de1dc27c94dbd651abb22..df61768f63e5df437067a4b64f870e7c973764f1 100755
--- a/PanACoTA/annotate_module/annotation_functions.py
+++ b/PanACoTA/annotate_module/annotation_functions.py
@@ -246,7 +246,7 @@ def prodigal_train(gpath, annot_folder):
                         logger=logger)
     prodigalf.close()
     prodigalferr.close()
-    if ret.returncode == 0:
+    if ret != 1 and ret.returncode == 0:
         logger.log(utils.detail_lvl(), f"End training on {gpath}")
         return gpath_train
     else:
diff --git a/containers/Dockerfile b/containers/Dockerfile
index d67e3997a9e41bf3b76ab46400389e5871ab69c2..542879d85ed2632bd1061e8fdbda46107ac04013 100644
--- a/containers/Dockerfile
+++ b/containers/Dockerfile
@@ -157,11 +157,11 @@ RUN tar -xzf iqtree-2.0.6-Linux.tar.gz &&\
     ln -s /install_dir/iqtree-2.0.6-Linux/bin/iqtree2 /usr/local/bin
 
 # # # Install PanACoTA
-# # WORKDIR /install-dir
-# # RUN wget https://github.com/gem-pasteur/PanACoTA/archive/refs/tags/1.4.0.tar.gz
-# # RUN tar -xzf v1.4.0.tar.gz
-# # WORKDIR /install-dir/PanACoTA-1.4.0
-# # RUN ./make
-
-# # ENTRYPOINT ["/usr/local/bin/PanACoTA"]
-# # CMD ['-h']
\ No newline at end of file
+WORKDIR /install-dir
+RUN wget https://github.com/gem-pasteur/PanACoTA/archive/refs/tags/v1.4.0.tar.gz
+RUN tar -xzf v1.4.0.tar.gz
+WORKDIR /install-dir/PanACoTA-1.4.0
+RUN ./make
+
+ENTRYPOINT ["/usr/local/bin/PanACoTA"]
+CMD ['-h']
\ No newline at end of file