When a piston move a slab with water, the water disappear (I think with stairs is the same)

The other day I was trying to make a mechanism and I put slabs with water that were moved by pistons, when they were moved the water disappeared automatically, please fix it so it can works.