If you remove the phrase "after the serve" and leave in "while the ball is live/in-play", you still have the same rule though. Taking this out and calling a fault for flicking a ball off the paddle for a serve would also introduce the side effect of having a fault called while the ball is no live/in-play which was changed in 2021 to simply when faults were called.