fixing sizing

fixed the sizing for the unit