diff --git a/Dockerfile b/Dockerfile
index 6f5e25a..0277f24 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -12,6 +12,7 @@ RUN git clone -b nvfp4-mega-moe https://sweetapi.com/biondizzle/DeepGEMM.git /ro
 
 # Build DeepGEMM with proper CUDA/NVRTC paths
 ENV CPATH="/usr/local/lib/python3.12/dist-packages/flashinfer/data/cutlass/include:/usr/local/lib/python3.12/dist-packages/nvidia/cu13/include:/usr/local/cuda-13.0/include:${CPATH}"
+ENV PYTHONPATH="/root/DeepGEMM:${PYTHONPATH}"
 # NVRTC lives in the pip nvidia/cu13 package, but the linker expects it in cuda/lib64
 # Create a symlink so -lnvrtc resolves
 RUN ln -sf /usr/local/lib/python3.12/dist-packages/nvidia/cu13/lib/libnvrtc.so.13 /usr/local/cuda/lib64/libnvrtc.so