Fix edit mode derp