It might be too strong for a wizard, but the request was to make it good for an EK. If you really want go that way, make TS only work on weapon attacks.
In any case, I think the situation for a wizard is not so clear as it sounds. Yes you could get advantage on your cantrip every round, but...