Loyalty trident not returning back after sticking into an extended piston arm

Loyalty trident isn't coming back when it sticks to an extended piston arm, but can return if the arm is retracted.

Steps to Reproduce:

1. Place a piston.
2. Activate the piston with a lever.
3. Throw a Loyalty trident at the extended piston arm.