Battery weight is a problem, but electric motors are also a lot lighter than diesels plus all the attendant plumbing and hardware. You don't even need that couple hundred pounds of drive shaft. Electrics are also far more efficient than diesels. This doesn't matter as much on long hauls, but regenerative braking provides a lot of charge with a heavy load.

This engineer did the math for the Tesla, and it checks out.