Fixed the flickering of screen in edit flow.