This isn't normally a problem with the blocks affected by the magnets by default, since they're just regular blocks. However, I've tried adding metal walls and stairs into the tag, but they would keep their form even after finishing. Another example is a metal block that has a separate blockstate for when it's powered. It will stay powered when pushed/pulled, and will only fix itself when updated.
Just wanted to let you know if you haven't heard already.