fixed edit mode class issue