diff --git a/main.sh b/main.sh
index b5d2840..4d6716e 100755
--- a/main.sh
+++ b/main.sh
@@ -12,6 +12,8 @@ VERSION="0.1"
 git clone https://github.com/pop-os/cosmic-comp.git
 cd cosmic-comp
 
+sed -i '/^\s*\(cargo\|rustc\|rust-all\)\s*\(([^)]*)\)\?\s*,\?$/d; s/,\s*\(cargo\|rustc\|rust-all\)\s*\(([^)]*)\)\?\s*,\?//g' ./debian/control
+sed -i '2iexport PATH := $(shell . /root/.cargo/env && echo $$PATH)\nexport $(shell . /root/.cargo/env && env | grep "^CARGO_\|^RUSTUP_" | sed "s/=/:=/")\n' ./debian/rules
 # Get build deps
 apt-get build-dep ./ -y
 apt-get install curl -y