diff --git a/.github/workflows/build_lint_test.yml b/.github/workflows/build_lint_test.yml
index d629f4c..7994982 100644
--- a/.github/workflows/build_lint_test.yml
+++ b/.github/workflows/build_lint_test.yml
@@ -22,7 +22,7 @@ jobs:
         docker run flowc-${{ matrix.dockerfile }} flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics
 
     - name: Test with pytest
-      run: docker run flowc-${{ matrix.dockerfile }} pytest --cov --cov-report=xml
+      run: docker run flowc-${{ matrix.dockerfile }} pytest /flowc --cov --cov-report=xml
 
     - name: Upload coverage reports to Codecov
       uses: codecov/codecov-action@v3
diff --git a/docker/Dockerfile-pytorch2.1.2 b/docker/Dockerfile-pytorch2.1.2
index fa08e64..3bd19d8 100644
--- a/docker/Dockerfile-pytorch2.1.2
+++ b/docker/Dockerfile-pytorch2.1.2
@@ -5,7 +5,7 @@ WORKDIR /app
 # Copy requirements and install dependencies
 
 # Copy the source code and install the package
-COPY . /app
-RUN pip install .
+COPY . /flowc
+RUN pip install /flowc
 
 CMD ["/bin/bash"]