Fixing returning from source to visual editor after edit