Issue with pistons

When you are in creative, when you break blocks you don't get them back. There are some issues when you do, but a main one is when if you place a piston (any) and extend it, and break it, you get it back.