diff --git a/README.md b/README.md
index 612d7c4..5a0a6bc 100644
--- a/README.md
+++ b/README.md
@@ -6,7 +6,7 @@ Build Python package:
 `python setup.py bdist_wheel`
 
 Build singularity recipe:
-`neurodocker generate singularity -b nvidia/cuda:10.2-cudnn7-devel-ubuntu18.04 -p apt --copy /home/tibuch/Gitrepos/FourierImageTransformer/dist/fourier_image_transformers-0.1.0-py3-none-any.whl /fourier_image_transformers-0.1.0-py3-none-any.whl --miniconda create_env=fit conda_install='python=3.7 astra-toolbox -c astra-toolbox/label/dev pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch' pip_install='/fourier_image_transformers-0.1.0-py3-none-any.whl' activate=true --entrypoint "/neurodocker/startup.sh python" > v0.1.0.Singularity`
+`neurodocker generate singularity -b nvidia/cuda:10.2-cudnn7-devel-ubuntu18.04 -p apt --copy /home/tibuch/Gitrepos/FourierImageTransformer/dist/fourier_image_transformers-0.1.0-py3-none-any.whl /fourier_image_transformers-0.1.0-py3-none-any.whl --miniconda create_env=fit conda_install='python=3.7 astra-toolbox pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch -c astra-toolbox/label/dev' pip_install='/fourier_image_transformers-0.1.0-py3-none-any.whl' activate=true --entrypoint "/neurodocker/startup.sh python" > v0.1.0.Singularity`
 
 Build singularity container:
 `sudo singularity build fit.simg fit.Singularity`
\ No newline at end of file
diff --git a/singularity/v0.1.0.Singularity b/singularity/v0.1.0.Singularity
index 442d042..b38c62a 100644
--- a/singularity/v0.1.0.Singularity
+++ b/singularity/v0.1.0.Singularity
@@ -1,6 +1,6 @@
 # Generated by: Neurodocker version 0.7.0
 # Latest release: Neurodocker version 0.7.0
-# Timestamp: 2020/12/29 15:21:22 UTC
+# Timestamp: 2020/12/29 15:40:35 UTC
 # 
 # Thank you for using Neurodocker. If you discover any issues
 # or ways to improve this software, please submit an issue or
@@ -53,14 +53,14 @@ conda create -y -q --name fit
 conda install -y -q --name fit \
     "python=3.7" \
     "astra-toolbox" \
-    "-c" \
-    "astra-toolbox/label/dev" \
     "pytorch" \
     "torchvision" \
     "torchaudio" \
     "cudatoolkit=10.2" \
     "-c" \
-    "pytorch"
+    "pytorch" \
+    "-c" \
+    "astra-toolbox/label/dev"
 sync && conda clean -y --all && sync
 bash -c "source activate fit
   pip install --no-cache-dir  \
@@ -102,14 +102,14 @@ echo '{
 \n        "conda_install": [
 \n          "python=3.7",
 \n          "astra-toolbox",
-\n          "-c",
-\n          "astra-toolbox/label/dev",
 \n          "pytorch",
 \n          "torchvision",
 \n          "torchaudio",
 \n          "cudatoolkit=10.2",
 \n          "-c",
-\n          "pytorch"
+\n          "pytorch",
+\n          "-c",
+\n          "astra-toolbox/label/dev"
 \n        ],
 \n        "pip_install": [
 \n          "/fourier_image_transformers-0.1.0-py3-none-any.whl"