Sticky Pistons does not pull chests back.

I was testing the new feature of pistons being able to move chests and I came across something. If I extend a sticky piston with a lever while holding a chest is in front of it, it will extend. But when I tried to pull the chest back, the piston got pulled back but the chest didn't get retracted with it.