There is a very big problem with the retraction of the piston, and that is that a mob is on it.(IMG1)

When it is activated, it does so without problems but when it retracts, it pulls the mob same.(IMG2-3)