diff --git a/dist/debian/build_deb.sh b/dist/debian/build_deb.sh
index edc7268..73c5ea2 100755
--- a/dist/debian/build_deb.sh
+++ b/dist/debian/build_deb.sh
@@ -100,12 +100,12 @@ cp dist/debian/rules.in debian/rules
 sed -i -e "s/@@VERSION@@/$SCYLLA_VERSION/g" debian/changelog
 sed -i -e "s/@@RELEASE@@/$SCYLLA_RELEASE/g" debian/changelog
 sed -i -e "s/@@CODENAME@@/$TARGET/g" debian/changelog
-if [ "$DISTRIBUTION" = "Ubuntu" ]; then
+if [ "$TARGET" = "trusty" ] || [ "$TARGET" = "xenial" ] || [ "$TARGET" = "yakkety" ] || [ "$TARGET" = "zesty" ] || [ "$TARGET" = "artful" ]; then
     sed -i -e "s/@@REVISION@@/0ubuntu1/g" debian/changelog
 else
     sed -i -e "s/@@REVISION@@/1/g" debian/changelog
 fi
-if [ "$RELEASE" = "14.04" ]; then
+if [ "$TARGET" = "trusty" ]; then
     sed -i -e "s/@@DH_INSTALLINIT@@/--upstart-only/g" debian/rules
 else
     sed -i -e "s/@@DH_INSTALLINIT@@//g" debian/rules