diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index e75e1ed028e359896ab595fc73368794d3f80ccf..2c2aca1279ce76ca25e80a631c2e344674524829 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -50,7 +50,9 @@ build:
   image: docker:20.10.16
   services:
     - name: docker:20.10.16-dind
-      command: ["--host=tcp://0.0.0.0:2375", "--tls=false", "--storage-driver=overlay2", "--mtu=1500", "--data-root=/mnt/volume"]
+      command: ["--host=tcp://docker:2375", "--tls=false", "--storage-driver=overlay2", "--mtu=1500", "--data-root=/mnt/volume"]
+      volumes:
+        - /var/run/docker.sock:/var/run/docker.sock
   script:
     - docker info
     - docker buildx create --use
@@ -69,4 +71,4 @@ deploy:
     - docker pull 22baar1mst/asyst-nlp-grader:latest
     - docker run -d 22baar1mst/asyst-nlp-grader:latest
   only:
-    - master
+    - master
\ No newline at end of file
diff --git a/flask_ml_api/Dockerfile b/flask_ml_api/Dockerfile
index 11de0b2993a9b91211287442019ace85866d73de..a09037025d8a4d34364584240da3b4fcafc880b9 100755
--- a/flask_ml_api/Dockerfile
+++ b/flask_ml_api/Dockerfile
@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime as build
+=FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime as build
 
 RUN apt-get update && apt-get install -y curl
 RUN curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs -o /usr/local/bin/rustup-init.sh