Fix issue #6 . Edit nodes not working well.