diff --git a/Dockerfile b/Dockerfile
index 601afb9..acabd29 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -40,7 +40,7 @@ FROM builder as build_image
 WORKDIR /builder
 #COPY --from=build_rootfs /build/rv64-sid/ ./rv64-port/
 COPY create_image.sh build.sh ./
-COPY rootfs/setup_rootfs.sh ./
+COPY rootfs/setup_rootfs.sh ./rv64-port/ 
 
 CMD /builder/build.sh ${KERNEL_VERSION}
 
diff --git a/create_image.sh b/create_image.sh
index 4da8582..f0252ec 100755
--- a/create_image.sh
+++ b/create_image.sh
@@ -83,8 +83,8 @@ function copy_rootfs(){
     get_kernel_deb
 
     cp -v /tmp/vf2_kernel_deb/*.deb ${ROOTFS_POINT}/tmp
-    cp -v /builder/setup_rootfs.sh ${ROOTFS_POINT}
-    chroot "${ROOTFS_POINT}" qemu-riscv64-static /bin/sh ${ROOTFS_POINT}/setup_rootfs.sh
+    cp -v /tmp/setup_rootfs.sh ${ROOTFS_POINT}
+    chroot "${ROOTFS_POINT}" qemu-riscv64-static /bin/sh /setup_rootfs.sh
     rm "${ROOTFS_POINT}/setup_rootfs.sh" "${ROOTFS_POINT}/usr/bin/qemu-riscv64-static"
 
     umount "${ROOTFS_POINT}"