Fix live site editing