Hitting a mob while swimming makes you stop swimming

What I expected to happen was...:
Keep swimming

What actually happened was...:
Hitting the mob makes you stop the swimming

Steps to Reproduce:
1. Stay underwater where have a mob;

2.  Start swimming (Ctrl + w)

3. Hit an mob (squid, fish, even a pig...) while swimming;

4. The bug appears here.

It appears on survival and creative mode, but I didn't test it in another game mode.