Optimize for edit mode!