unit.sh fixes