Yes a TL01 would do the trick
But if you're used to driving a TA04, you won't like the TL. However, you could go for an even better chassis, the TB01. Make sure it's a rally-version, the street-versions come without dustproof sponges here and there, and there's no bathtub lid. And get a "full ball bearing version" aswell, it saves having to buy them on the side.